基于GSR-YOLOv7的探地雷达回波图像识别的研究

作者:钟运杰; 李康; 莫思特*; 李碧雄
来源:半导体光电, 2023, 44(05): 767-774.
DOI:10.16818/j.issn1001-5868.2023061401

摘要

针对探地雷达回波图像特征不突出,工作人员现场解译时间长,准确率不高的问题,提出一种基于YOLOv7改进的回波图像识别方法GSR-YOLOv7。首先使用GhostConv替换YOLOv7卷积层的卷积核以减小参数量;引入SimAM注意力机制提高特征学习能力;通过对感受野模块RFB卷积核的复用,增大感受野;同时引入改进EIoU损失函数提高模型分类能力和回归精度。基于道路回波图像数据集的实验结果表明,改进后的模型MAP50达到了97.57%,MAP50∶95达到了73.13%,较YOLOv7分别提高2.13%和8.46%,模型大小减小了35%。所提出的GSR-YOLOv7模型对于回波目标的检测效果较好,适用于移动端系统,对于体积小、算力低的平台具有较大应用价值。

全文