Yaoze Sun, Junwei Gao. Defect Detection of Wheel Set Tread Based on Improved YOLOv5[J]. Laser & Optoelectronics Progress, 2022, 59(22): 2215003
- Laser & Optoelectronics Progress
- Vol. 59, Issue 22, 2215003 (2022)
Fig. 1. YOLOv5 model structure
Fig. 2. Structure of convolution attention mechanism
Fig. 3. Structure diagram of improved YOLOv5
Fig. 4. Tread defect type. (a) Normal; (b) scratch; (c) laceration injury; (d) peel
Fig. 5. Loss variation curve
Fig. 6. Comparison of detection results between YOLOv5 and improved YOLOv5. (a) (c) (e) Detection results of YOLOv5; (b) (d) (f) detection results of the improved YOLOv5
Fig. 7. Comparison of inclusion detection results. (a) Scratch; (b) including scratch; (c) confluent scratch
|
Table 1. Target quantity in data
|
Table 2. Experimental results of different improvements
|
Table 3. Comparison of model complexity
|
Table 4. Performance comparison of mainstream target detection algorithms
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20