基于雙通道Transformer的地鐵站臺異物檢測
計算機工程
頁數(shù): 11 2023-07-31
摘要: Transformer因其全局注意力優(yōu)勢在異物檢測上取得了比卷積神經(jīng)網(wǎng)絡(luò)(CNN)更具競爭力的結(jié)果,但依然面臨計算成本高、輸入圖像塊尺寸固定、局部與全局信息交互匱乏等問題。提出一種基于雙通道Transformer骨干網(wǎng)絡(luò)、金字塔輕量化Transformer塊和通道交叉注意力機制的DualF ormer模型,用以檢測地鐵站臺屏蔽門與列車門間隙中存在的異物。針對輸入圖像塊尺寸固定的...