摘要:隨著(zhù)社會(huì )經(jīng)濟的發(fā)展,人們的生活水平持續提高,生活垃圾量急劇攀升。為了有效應對垃圾分揀效率低、準確率差等問(wèn)題,提出一種以YOLOv7網(wǎng)絡(luò )為基礎模型的垃圾檢測算法。該算法對YOLOv7網(wǎng)絡(luò )進(jìn)行了一系列改造,首先,在Head模塊添加了注意力機制SimAM,增強了模型的感知能力和自適應能力,從而提高檢測精度;其次,在主干網(wǎng)絡(luò )中改進(jìn)了非極大值抑制算法(soft-NMS)去除冗余的檢測框,再次改進(jìn)了損失函數為邊框回歸損失函數SIoU,提高了檢測的精度和速度;最后,采用C3模塊替換YOLOv7有的ELAN-W模塊,提升網(wǎng)絡(luò )對較小目標的檢測能力。通過(guò)數據集對改進(jìn)的網(wǎng)絡(luò )進(jìn)行測試,平均準確度為98.93%、訓練時(shí)間為27.58h,高于原模型的96.31%、44.53h,實(shí)驗結果也表明改進(jìn)算法的檢測效果有較為明顯的提升。