Road pothole detection based on improved YOLOv7

Abstract

According to the World Health Organization, the current global death toll from road traffic accidents is as high as 1.3 million annually. The main cause of road traffic accidents is poor road conditions, and potholes on roads are the most serious type of road diseases. Therefore, timely detection and treatment of road potholes is very necessary. This paper proposes a method based on the use of YOLOv7 deep learning model to detect potholes on the road. At the same time, CBAM attention mechanism and optimization of loss function are added on the basis of this method. Combined with the idea of transfer learning, the improved YOLOv7 network is trained. The final test results are significantly improved compared with other road potholes detection models. F1 score is 78%, Precision value can reach 85.81%, and mAP value can reach 83.02%.