湖南大学学报（自然科学版）2024，Vol.51Issue(12)：107-118,12.DOI:10.16339/j.cnki.hdxbzkb.2024288
多病害并发复杂场景下的道路病害检测RGT-YOLOv7模型
Road Disease Detection RGT-YOLOv7 Model under Multiple Diseases Complicated Scenarios
摘要
Abstract
With the continuous expansion of China's road network,road disease detection has become an indispensable part of road maintenance and traffic safety,and road disease detection based on deep learning has become a research hotspot in this field.Aiming at the problems of low accuracy and generalization ability of road disease identification in complex scenes with multiple diseases,a road disease detection model called Receptive Ghost Triplet-YOLOv7(RGT-YOLOv7)in complex scenes is proposed in this paper.A triplet attention mechanism is introduced in the backbone network to improve the correlation of disease features in different channels and spaces,and to solve the problem of low feature extraction efficiency.The original SPP module is replaced by the SPPF module,the Ghost module is added to improve the utilization rate of redundant features,and the original redundant features and the newly extracted features are fused to get more diverse and rich feature information with different scales.In order to improve the model perception field,RFBs module is added in the feature enhancement part,and the feature map is extracted from different directions by using cavity convolution with different sizes to enhance the extraction of horizontal and vertical features.Experimental results show that the average accuracy and balanced F score are improved by 6.9 percentage points and 3.9 percentage points,respectively,compared with YOLOv7,especially the longitudinal fracture identification is improved by 22.3 percentage points,and it also has good performance improvement compared with Faster R-CNN,YOLOv5,and recently proposed algorithm models.It is an effective road disease detection algorithm for proposed RGT-YOLOv7 under complex scenes.关键词
目标检测/道路病害检测/深度学习/YOLOv7/RFBKey words
object detection/road distress detection/deep learning/YOLOv7/receptive field block(RFB)分类
信息技术与安全科学引用本文复制引用
罗向龙,王彦博,蒲亚亚,刘若辰..多病害并发复杂场景下的道路病害检测RGT-YOLOv7模型[J].湖南大学学报（自然科学版）,2024,51(12):107-118,12.基金项目
国家自然科学基金资助项目(62001059),National Natural Science Foundation of China(62001059) （62001059）
陕西省自然科学基金资助项目(2022JM-056),Natural Science Foundation of Shaanxi Province(2022JM-056) （2022JM-056）
