Research on Train Carriage Number Location Based on Region Segmentation Using PSENet
Aiming at the problem of high missed detection rate caused by the serious distortion of the existing text loca-tion methods for the carriage number of running trains in natural scenes,an accurate location method was proposed for the carriage number of trains based on region segmentation.The network model used in this method is based on PSENet.First,the ResNet50 network with channel domain attention was used to extract features,to cause the model to focus more on convolution channel information,to redistribute the weights from the perspective of channel domain,so as to im-prove the positioning accuracy of carriage number.Then the feature pyramid and bottom-up path enhancement module were used to fuse multi-scale feature maps,where the strong positioning features in the shallow network were propagated to the deep network to accurately locate the carriage number area from the complex environment,which is conducive to reducing the missing rate of carriage numbers.Finally,the gradual scale expansion module based on the breadth first al-gorithm was used to expand and segment the fused feature map from small scale to large scale,while the loss function combined with the set similarity measure Dice coefficient was used to classify and regress the segmentation results and output the positioning results.The research results show that,through training and verification on the real train carriage number image dataset,the network model proposed in this paper achieves a positioning precision of 97.47％,a recall rate of 94.14％,a comprehensive F1 score of 95.78％,and a prediction time of about 0.2 seconds for a single image.Compared to the basic PSENet,the positioning accuracy of the carriage number has increased by 3.78％,with the in-crease of the recall rate by 1.71％.Overall,the F1 score of the car number has increased by 2.73％.The research re-sults can provide a high-precision detection and location method for the automatic detection and identification system of train carriage numbers.
freight traintrain carriage number locationPSENetchannel attentionmulti-scale feature fusion
万方数据
维普
