Abstract
Surface defect detection of crane wire rope is very important for safe operations of crane. However, the surface defect characteristics of crane wire rope are varied, there is no standardized data and tiny-size, which makes the universal surface defect detection models ineffective. Therefore, a surface defect detection model named TBi-YOLOv5 is proposed in this article. First, a random augment method is used to search the optimal data augment strategy automatically and enforce the generalization ability of the surface defect detection model. Then, in order to enhance the feature extraction ability on both the local and global views, a Bottleneck Transformer (BOT) module is added in YOLOv5, which makes effective fusion of convolutional neural network (CNN) and multi-head self-attention mechanism (MHSA). After that, inspired by the idea of bidirectional feature pyramid network (BiFPN), a small target detection layer (STPL) is added in the neck part of YOLOv5 to detect the tiny-size surface defect on the wire rope. Experiments on the surface defect detection of crane wire rope show the effectiveness of the proposed TBi-YOLOv5. Compared with YOLOv5s, the mean average precision (mAP) of TBi-YOLOv5 has been increased by nearly 4%.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
More From: Proceedings of the Institution of Mechanical Engineers, Part C: Journal of Mechanical Engineering Science
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.