The invention discloses a network-based rail B-display image recognition method, which belongs to the field of railway track flaw detection and
deep learning, and comprises the following steps: 1, acquiring a rail B-display image, dividing the B-display image into five types of labels, namely a rail end
label, a rail head nuclear flaw
label, a normal screw hole
label, an abnormal screw hole label and a lead hole label, and constructing a
data set after performing data enhancement on
small sample types; 2, a
data set is put into the model for training, an attention module is embedded into the model, the sensing ability of key features is enhanced through channel re-calibration, and the detection precision of
small target damage is improved; and 3, aiming at a thin and long target with an extreme
aspect ratio, an EIoU
loss function is adopted to optimize bounding box regression, and the positioning precision and the convergence speed of the model in a complex form are remarkably improved. According to the invention, fine defects such as rail head nuclear injury, abnormal screw holes and the like can be effectively identified, the omission ratio and the
false detection ratio are reduced, intelligent identification and automatic analysis of B-display images are realized, the manual dependence is greatly reduced, and the detection efficiency and the railway
operation safety are improved.