改進(jìn)YOLOv7的X光圖像危險(xiǎn)品檢測算法

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