PDF(2178 KB)
改进YOLOv5的复杂场景下水泥路面病害检测
张在岩, 宋伟东, 邬嘉晨
PDF(2178 KB)
PDF(2178 KB)
改进YOLOv5的复杂场景下水泥路面病害检测
Disease detection of cement pavement based on improved YOLOv5 in complex scenarios
针对国内水泥路面病害检测数据集缺乏、规模小、场景单一,以及深度学习算法在复杂场景下泛化能力不足的问题,提出一种基于改进YOLOv5的路面病害检测算法。收集并构建包含11 862张图像的水泥路面病害检测数据集,覆盖9类场景下的3类最常见病害类型;通过融合以IoU度量的 K-Means 聚类算法和遗传算法获取模型训练的先验锚框;在特征增强阶段,引入轻量级上采样模块(CARAFE),减少特征重组过程中的信息损失;引入顾及通道、高度和宽度维度的多维协同注意力模块(MCA),增强多尺度病害特征的辨别力。实验结果表明:所提算法在保持较快推理速度的前提下,F 1分数和平均精确率(mAP)分别达到75.5%和81.6%,优于5种主流的目标检测算法。实例分析表明:基于改进YOLOv5的路面病害检测算法能够满足大规模水泥路面病害智能检测与破损状况评价的实际需求。
Aiming at the problems of lack of domestic cement pavement disease detection data sets, small scale, single scene, and insufficient generalization ability of deep learning algorithms in complex scenes, a pavement disease detection algorithm based on improved YOLOv5 is proposed. A cement pavement disease detection dataset containing 11 862 images was collected and constructed, covering 3 most common disease types in 9 scenarios. The prior anchor frame of model training is obtained by combining the K-Means clustering algorithm measured by IoU and genetic algorithm. In the feature enhancement stage, a lightweight upsampling module (CARAFE) is introduced to reduce the information loss in the feature recombination process. A multi-dimensional collaborative attention module (MCA) considering channel, height and width dimensions is introduced to enhance the discrimination of multi-scale disease features. The experimental results show that under the premise of maintaining fast inference speed, the average of F 1 score and average precision (mAP) of the proposed algorithm reach 75.5% and 81.6%, respectively, which are better than the five mainstream target detection algorithms. The example analysis shows that the pavement disease detection algorithm based on improved YOLOv5 can meet the actual needs of large-scale cement pavement disease intelligent detection and damage condition evaluation.
水泥路面 / 深度学习 / ISTD-PDD3数据集 / 病害检测 / ISTD-YOLO模型
cement pavement / deep learning / ISTD-PDD3 data set / disease detection / ISTD-YOLO model
P258
| 1 |
|
| 2 |
晏班夫,徐观亚,栾健,等.基于Faster R-CNN与形态法的路面病害识别[J].中国公路学报,2021,34(9):181-193.
|
| 3 |
|
| 4 |
|
| 5 |
|
| 6 |
|
| 7 |
|
| 8 |
|
| 9 |
|
| 10 |
李章维,胡安顺,王晓飞.基于视觉的目标检测方法综述[J].计算机工程与应用,2020,56(8):1-9.
|
| 11 |
|
| 12 |
|
| 13 |
|
| 14 |
|
| 15 |
罗晖,贾晨,李健.基于改进YOLOv4的公路路面病害检测算法[J].激光与光电子学进展,2021,58(14):336-344.
|
| 16 |
|
| 17 |
|
| 18 |
|
| 19 |
惠冰,郭牧,王洲,等.基于三维激光技术的路面坑槽多维度指标检测[J].同济大学学报(自然科学版),2018,46(1):60-66.
|
| 20 |
郭小英,李文书,钱宇华,等.可计算图像复杂度评价方法综述[J].电子学报,2020,48(4):819-826.
|
| 21 |
|
| 22 |
|
| 23 |
|
| 24 |
|
| 25 |
|
| 26 |
|
| 27 |
|
| 28 |
|
| 29 |
|
| 30 |
|
| 31 |
|
| 32 |
徐鹏,祝轩,姚丁,等.沥青路面养护智能检测与决策综述[J].中南大学学报(自然科学版),2021,52(7):2099-2117.
|
/
| 〈 |
|
〉 |