嵌入最大平均池化SE模块的YOLOv7对道路落石的检测

DETECTIONOFROADROCKFALLBYYOLOV7EMBEDDEDINMAXIMUMAVERAGEPOOLINGSEMODULE

  • 摘要: 为帮助在复杂多变的山区道路中驾车行驶的驾驶人员及时发现行驶前方的落石,借鉴"聚类方法"以一种创新方式完成落石标注,将全局最大平均池化嵌入到SE注意力机制完成MA-SE模块的设计,将MA-SE模块在YOLOv7网络中以模块群方式拓展到ELAN模块、基于区位优势将其填充到SPPCSPC模块的同时将IOU损失函数进行并联计算优化,提出基于嵌入最大平均池化SE模块的新型YOLOv7网络模型。在自制落石数据集上,改进的YOLOv7算法在FPS几乎未减少的情况下将mAP@0.5提升1.6百分点。目前山区道路落石的检测研究较少,此研究具有一定的研究价值。

     

    Abstract: In order to help drivers on complex and variable mountain roads timely discover falling rocks in front of them, the "clustering method" was used to mark falling rocks in an innovative way, and the global maximum average pooling was embedded into the SE attention mechanism to complete the design of MA- SE module. The MA- SE module was extended to ELAN module in YOLOv7 network by means of module group, and filled into SPPCSPC module based on location advantage. Meanwhile, the IOU loss function was optimized in parallel, and a new YOLOv7 network model based on embedded maximum average pooling SE module was proposed. On the self- made rockfall data set, the improved YOLOv7 algorithm increases mAP@0.5 by 1.6% with almost no decrease in FPS. At present, there are few studies on the detection of rockfall on mountain roads, which has certain research value.

     

/

返回文章
返回