计算机科学
特征提取
编码器
人工智能
目标检测
变压器
特征(语言学)
计算机视觉
模式识别(心理学)
机器学习
工程类
电压
语言学
哲学
电气工程
操作系统
作者
Yu‐Chen Liu,Fang Liu,Wei Liu,Yucheng Huang
出处
期刊:IEEE Transactions on Intelligent Transportation Systems
[Institute of Electrical and Electronics Engineers]
日期:2023-01-01
卷期号:: 1-10
被引量:5
标识
DOI:10.1109/tits.2023.3306578
摘要
Timely and accurately detection as well as rehabilitation of road surface defects are of utmost importance for ensuring road safety and minimizing maintenance cost. However, the variety of pavement distress types and forms makes it difficult to accurately classify and detect them. To tackle the issue, this paper proposes a novel target detection model YOLO-SST based on YOLOv5 with the improvement in pavement distress features. First, a Shuffle Attention mechanism is introduced in the feature extraction backbone network to enhance the detection ability without significantly increasing the computational cost. Secondly, we add a detection layer and embed Swin-Transformer encoder blocks into the C3 module to capture global and contextual information. Finally, to improve the model’s detection ability, transfer learning is employed on a self-made dataset called RDDdect_2023, which consists of street view images captured via a DJI Action camera mounted on the car. Experimental results demonstrate that the YOLO-SST model outperforms YOLOv5 and other target detection models in terms of accuracy, recall rate, and mAP@0.5 value for detecting pavement distresses. This confirms that the YOLO-SST model has stronger feature extraction and fusion capabilities, resulting in better detection performance.
科研通智能强力驱动
Strongly Powered by AbleSci AI