Aiming at the problem that the multi-scale feature interaction ability of the traditional deep learning-based line selection algorithm is insufficient, resulting in the decline of line selection accuracy, a multi-scale feature fusion line selection method based on transfer learning is proposed, abbreviated as TLM-Net. Firstly, to address the issue of the insufficient generalization ability of the line selection network in small-sample scenarios, a simulation data pre-training framework is constructed, and a robust feature representation basis is established through a cross-domain knowledge transfer mechanism. Secondly, aiming at the problem of insufficient extraction of feature information by traditional algorithms, a multi-scale feature fusion network (MFFN) is designed to integrate global context information and local detail features, achieving cross-level semantic complementarity and spatial alignment optimization. Then, to enhance the representation ability of weak fault feature information, an EKA mechanism integrating variable kernel convolution is designed. The background interference is reduced through adaptive multi-region feature focusing, and the edge recognition accuracy of the model for irregular targets is improved. Finally, the pre-trained model is transferred to the target domain by adopting the transfer learning strategy, and the network parameters are fine-tuned in combination with the on-site data to achieve cross-domain adaptation of the feature space. The experimental results show that the TLM-Net algorithm’s mAP@0.5 reaches 98.5%, the accuracy rate and recall rate reach 98.3% and 96.5%, respectively, and the accuracy is improved by 37.5% compared with the original model.
Li et al. (Wed,) studied this question.