Real-time sewer defect detection based on YOLO network, transfer learning, and channel pruning algorithm

Zuxiang Situ,Shuai Teng,Xiaoting Liao,Gongfa Chen,Qianqian Zhou
DOI: https://doi.org/10.1007/s13349-023-00681-w
2023-02-01
Journal of Civil Structural Health Monitoring
Abstract:Finding an accurate, rapid, and automated real-time detection method for sewer is of great significance to improve the detection efficiency in sewer maintenance and ensure the system functionality. Deep learning technologies developed in recent years offered novel alternatives, however, most of which are computation-intensive, limiting their applications to small mobile terminals. Focusing on the characteristics and detection requirements of sewer defects, this study proposed a real-time detection method based on an integration of YOLO_v5-based object detection network, transfer learning, and channel pruning technique. Results showed that the method significantly improved the detection accuracy and speed. With the aid of model pruning, the number of parameters and operations of the original model were reduced by 81.0% and 48.8%, respectively. The model superiority was further verified by comparing with nine mainstream detection algorithms and four lightweight backbone networks. It was concluded that our model had great advantages in practical engineering applications and projects. The knowledge gained provided technical references for the deployment of portable devices and sewer inspection robots in the future.
engineering, civil
What problem does this paper attempt to address?