Abstract

There are many kinds of pavement diseases, and the detection background in the actual scene is complex, so the accuracy of the detection model is not high. Based on YOLOv5s, a pavement diseases detection model YOLOv5s-CBP is proposed, and the CBAM (Convolutional Block Attention Module) attention module was introduced into the backbone network to enhance the model's attention to the characteristics of pavement diseases. The BiFPN network was constructed in the neck, which realizes the cross-layer exchange and integration of information and retains more high-level semantic information. A small target detection layer P2 was added to the detection head to improve the detection ability of the model for small road diseases. The experimental results on data sets show that the precision and recall of the model are improved by 6.1% and 4.4% respectively, and the mAP is improved by 5.7%, which effectively improves the performance of the detection model.

Talk to us

Join us for a 30 min session where you can share your feedback and ask us any queries you have

Schedule a call

Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.