改進(jìn)YOLOv5的X光圖像違禁品檢測算法

摘要： 針對(duì)X光圖像違禁品檢測中的復(fù)雜背景、正負(fù)類別不平衡和漏檢等問題，提出一種基于YOLOv5的X光違禁品檢測算法。該算法通過在YOLOv5s骨干網(wǎng)絡(luò)中引入Swin Transformer模塊，利用局部自注意力與Shifted Window機(jī)制提升模型對(duì)X光圖像全局特征的提取能力，并且在主干網(wǎng)絡(luò)后增加空間注意力機(jī)制與通道注意力機(jī)制，以提升算法對(duì)違禁品關(guān)鍵特征的提取能力。引入一種自適應(yīng)...