计算机工程与科学

作者：霍爱清, 李易

单位：（西安石油大学电子工程学院,陕西 西安 710065）

关键词：箭头标识线检测,M2Det,Mish激活函数,Mosaic数据增强,平均准确率,

基金：陕西省教育厅基金（17JS108）；陕西省科技厅一般工业项目（2020GY-152）；西安石油大学研究生创新与实践能力培养项目（YCS20213159)

针对M2Det算法在地面箭头标识线检测时存在准确率低、参量大等问题，提出了一种改进M2Det算法。该算法在特征提取时采用改进的主干特征提取网络和多级金字塔网络，利用非极大抑制对生成的密集边界框和类别分数进行筛选，进而获得检测结果。改进的M2Det算法用MobileNet v1轻量级网络替换VGG网络，用以减少参量；用Mish激活函数替换ReLU激活函数，同时在MobileNet v1网络中增加BasicRFB模块，用以提高检测精度；还引入Mosaic数据增强以实现数据扩充。实验数据集采用自主标注的地面箭头标识线构造，实验结果表明，改进的M2Det算法在地面箭头标识线检测中mAP达到88.72%，相比M2Det算法提升了约3.9个百分点，也明显高于其它对比算法。

来源：2022年第6期

《计算机工程与科学》期刊编辑部

查看计算机工程与科学杂志2022年第6期