基于改進(jìn)YOLO v5s的溫室番茄檢測模型輕量化研究
江蘇農(nóng)業(yè)科學(xué)
頁數(shù): 10 2024-04-20
摘要: 番茄檢測模型的檢測速度和識(shí)別精度會(huì)直接影響到番茄采摘機(jī)器人的采摘效率,因此,為實(shí)現(xiàn)復(fù)雜溫室環(huán)境下對(duì)番茄精準(zhǔn)實(shí)時(shí)的檢測與識(shí)別,為采摘機(jī)器人視覺系統(tǒng)研究提供重要的參考價(jià)值,提出一種以YOLO v5s模型為基礎(chǔ),使用改進(jìn)的MobileNet v3結(jié)構(gòu)替換主干網(wǎng)絡(luò),平衡模型速度和精度。同時(shí),在頸部網(wǎng)絡(luò)引入Ghost輕量化模塊和CBAM注意力機(jī)制,在保證模型檢測精度的同時(shí)提高模型的檢測速...