Comparative Analysis of Improved YOLO v5 Models for Corrosion Detection in Coastal Environments

Abstract

Coastal areas face severe corrosion issues, posing significant risks and economic losses to equipment, personnel, and the environment. YOLO v5, known for its speed, accuracy, and ease of deployment, has been employed for the rapid detection and identification of marine corrosion. However, corrosion images often feature complex characteristics and high variability in detection targets, presenting significant challenges for YOLO v5 in recognizing and extracting corrosion features. To improve the detection performance of YOLO v5 for corrosion image features, this study investigates two enhanced models: EfficientViT-NWD-YOLO v5 and Gold-NWD-YOLO v5. These models specifically target improvements to the backbone and neck structures of YOLO v5, respectively. The performance of these models for corrosion detection is analyzed in comparison with both YOLO v5 and NWD-YOLO v5. The evaluation metrics including precision, recall, F1-score, Frames Per Second (FPS), pre-processing time, inference time, non-maximum suppression time (NMS), and confusion matrix were used to evaluate the detection performance. The results indicate that the Gold-NWD-YOLO v5 model shows significant improvements in precision, recall, F1-score, and accurate prediction probability. However, it also increases inference time and NMS time, and decreases FPS. This suggests that while the modified neck structure significantly enhances detection performance in corrosion images, it also increases computational overhead. On the other hand, the EfficientViT-NWD-YOLO v5 model shows slight improvements in precision, recall, F1-score, and accurate prediction probability. Notably, it significantly reduces inference and NMS time, and greatly improves FPS. This indicates that modifications to the backbone structure do not notably enhance corrosion detection performance but significantly improve detection speed. From the application perspective, YOLO v5 and NWD-YOLO v5 are suitable for routine corrosion detection applications. Gold-NWD-YOLO v5 is better suited for scenarios requiring high precision in corrosion detection, while EfficientViT-NWD-YOLO v5 is ideal for applications needing a balance between speed and accuracy. The findings can guide decision making for corrosion health monitoring for critical infrastructure in coastal areas.

Keywords

• marine corrosion detection
• YOLO v5
• model improvement
• detection performance metrics
• comparative analysis