Accurately identifying the connecting handle of the train coupler is of great significance for the operation of the uncoupling robot. A train connection handle target recognition algorithm based on improved YOLOv5 was proposed to address this issue. The C_switchable atrous convolution(C_SAC) module was integrated into the backbone feature extraction network, and the wise intersection over union(WIOU) function was introduced as a new bounding box loss function to enhance the feature extraction ability of the backbone network, improve the model’s generalization ability and convergence rate. Then, images of the connecting handles of train couplers in different environments and positions on the production site were collected for recognition. The experimental results show that the improved YOLOv5 algorithm achieves a target recognition rate of 96.6% for the connecting handles of train couplers. Compared with the original algorithm, it shows significant improvements in accuracy, recall, average accuracy, and other aspects. Finally, it was applied in the development of an automatic uncoupling robot for train carriages, greatly improving the accuracy and effectiveness of automatic uncoupling.
| 科 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 |