The internal damage of the steel rail is serious, but the non-destructive testing B-display detection image has a lot of noise and noise, and the spatiotemporal distribution characteristics of different damages are not obvious, making it difficult to effectively identify. In response to this situation, a rail screw hole crack B-image recognition algorithm based on improved YOLOv8 was studied to improve the accuracy of intelligent identification of rail damage. Firstly, to reduce the missed detection of small damage targets, RepHGNetv2 network was used to optimize the YOLOv8 backbone network and improve the detection recall rate. Then, in order to improve the adaptability of the model to different types of damage detection, the detection head of YOLOv8 was replaced with Effientnet to improve the detection accuracy of the model. Finally, the LSKA attention mechanism was introduced into the SPPF module to enhance the model’s anti-interference ability against noise signals and improve its accuracy. The actual line detection results have verified that the detection accuracy of the above model reaches 95.1%, the recall rate reaches 93.8%, and the average accuracy reaches 97.6%, which is improved compared to other commonly used algorithms.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |