基于改進DINO的鐵路接觸網(wǎng)異物檢測方法
中國鐵道科學
頁數(shù): 10 2024-07-15
摘要: 針對鐵路接觸網(wǎng)異物類別多樣、場景多變的開放目標檢測問題,提出基于改進DINO目標檢測算法的鐵路接觸網(wǎng)異物檢測方法。首先,基于鐵路接觸網(wǎng)異物圖像特征,利用EfficientNet網(wǎng)絡替換原始模型中的Resnet主干網(wǎng)絡,并結合卷積注意力模塊(Convolutional Block Attention Module,CBAM)對EfficientNet網(wǎng)絡進行改進,在頸部結構采用改...