改進YOLOv7的X光圖像危險品檢測算法
計算機工程與應(yīng)用
頁數(shù): 10 2023-11-29
摘要: 針對X光安檢圖像在危險品檢測時背景復(fù)雜、遮擋嚴重、尺度多變等問題,對YOLOv7算法進行了改進,在提高檢測精度的同時使網(wǎng)絡(luò)更加輕量化。首先構(gòu)建PS-ELAN模塊替換原主干網(wǎng)絡(luò)中的ELAN模塊,減少網(wǎng)絡(luò)計算量和內(nèi)存占用,同時提升網(wǎng)絡(luò)的特征提取能力。其次將無參注意力機制SimAM與可變形卷積DCNv2融合至頸部網(wǎng)絡(luò)的下采樣階段,提高網(wǎng)絡(luò)對X光圖像危險品關(guān)鍵特征的捕捉能力。最后引入D... (共10頁)