改进型DSSD算法在道路损伤检测中的应用研究
2021年电子技术应用第12期
苏 可1，郭学俊2，杨 莹3，陈泽华2
1.太原理工大学 电气与动力工程学院，山西 太原030024； 2.太原理工大学 大数据学院，山西 晋中030600；3.山西省交通科技研发有限公司，山西 太原030006
摘要： 在自动检测中，由于道路损伤数据集存在小目标损伤难检测与类别不平衡问题，导致道路损伤检测的准确率低、虚假率高。为此，在DSSD(Deconvolutional Single Shot Detector)网络模型的基础上，提出一种结合注意力机制和Focal loss的道路损伤检测算法。首先，采用识别精度更高的ResNet-101作为DSSD模型的基础网络；其次，在ResNet-101主干网络中添加注意力机制，采用通道域注意力和空间域注意力结合的方式，实现特征在通道维度上的加权与空间维度上的聚焦，提升对小目标道路损伤的检测效果；最后，为了减少简单样本的权重，增大难分类样本的权重，使用Focal loss来提高整体的检测效果。在Global Road Damage Detection Challenge比赛所提供的数据集上进行验证，实验结果表明，该模型的平均精度均值为83.95%，比基于SSD和YOLO网络的道路损伤检测方法的准确率更高。
中图分类号： TP391.41
文献标识码： A
DOI：10.16157/j.issn.0258-7998.211684
中文引用格式： 苏可，郭学俊，杨莹，等. 改进型DSSD算法在道路损伤检测中的应用研究[J].电子技术应用，2021，47(12)：64-68，99.
英文引用格式： Su Ke，Guo Xuejun，Yang Ying，et al. Research on application of improved DSSD algorithm in road damage detection[J]. Application of Electronic Technique，2021，47(12)：64-68，99.
文献标识码： A
DOI：10.16157/j.issn.0258-7998.211684
中文引用格式： 苏可，郭学俊，杨莹，等. 改进型DSSD算法在道路损伤检测中的应用研究[J].电子技术应用，2021，47(12)：64-68，99.
英文引用格式： Su Ke，Guo Xuejun，Yang Ying，et al. Research on application of improved DSSD algorithm in road damage detection[J]. Application of Electronic Technique，2021，47(12)：64-68，99.
Research on application of improved DSSD algorithm in road damage detection
Su Ke1，Guo Xuejun2，Yang Ying3，Chen Zehua2
1.College of Electrical and Power Engineering，Taiyuan University of Technology，Taiyuan 030024，China； 2.College of Data Science，Taiyuan University of Technology，Jinzhong 030600，China； 3.Shanxi Transportation Technology Research and Development Co.，Ltd.，Taiyuan 030006，China
Abstract： In the automatic detection, the road damage data set has the problems of difficult detection of small target damage and imbalance of categories, resulting in low accuracy and high false rate of road damage detection. For this reason, based on the DSSD(deconvolutional single shot detector) network model, a road damage detection algorithm combining attention mechanism and Focal loss is proposed. First of all, ResNet-101 with higher recognition accuracy is used as the basic network of the DSSD model. Secondly, an attention mechanism is added to the ResNet-101 backbone network, and the channel domain attention and spatial domain attention are combined to achieve the weighting of features in the channel dimension and the focus on the spatial dimension, and improve the detection effect of small target road damage. Finally, in order to reduce the weight of simple samples and increase the weight of difficult-to-classify samples, Focal loss is used to improve the overall detection effect. It is verified on the data set provided by the Global Road Damage Detection Challenge competition. The experimental results show that the average accuracy of the model is 83.95%, which is more accurate than the road damage detection method based on SSD and YOLO network.
Key words : road damage detection；DSSD target detection algorithm；small target detection；attention mechanism；category imbalance problem
0 引言
道路建设是衡量国家现代化水平的重要指标之一，我国道路交通网庞大复杂，道路养护问题凸显。如何从道路图像快速准确地检测出损伤区域及类型成为学者研究的热点。
随着深度学习的快速发展，使用卷积神经网络(Convolution Neural Network，CNN)[1]自主地从数据集中提取相应特征信息成为主流方法，如快速的R-CNN[2]、SSD[3]、YOLO[4]等。这些网络能够定位和识别图中具有边界框的对象，为复杂背景下道路检测提供了有效的框架。
本文详细内容请下载：http://www.chinaaet.com/resource/share/2000003873。
作者信息：
苏 可1，郭学俊2，杨 莹3，陈泽华2
(1.太原理工大学 电气与动力工程学院，山西 太原030024；
2.太原理工大学 大数据学院，山西 晋中030600；3.山西省交通科技研发有限公司，山西 太原030006)
此内容为AET网站原创，未经授权禁止转载。