基于改進(jìn)YOLOv7-tiny的多光譜蘋果表層缺陷檢測
激光與光電子學(xué)進(jìn)展
頁數(shù): 9 2023-11-09
摘要: 針對蘋果表層存在多種缺陷類型、對不同缺陷的檢測方法不同的問題,提出一種基于改進(jìn)YOLOv7-tiny的缺陷檢測模型,結(jié)合相機(jī)采集的RGB+NIR多光譜圖像對蘋果表層多種缺陷進(jìn)行了檢測和分類。首先,為了提取更多有效的特征信息,提高對缺陷的定位能力,在主干網(wǎng)絡(luò)中使用坐標(biāo)注意力(CA)機(jī)制聚合坐標(biāo)信息,同時在主干網(wǎng)絡(luò)后添加上下文轉(zhuǎn)換器(CoT)模塊以增加全局感受野;其次,為了增強(qiáng)高效...