基于改進輕量化YOLO v7的成熟期香梨目標檢測方法
打開文本圖片集
doi:10.15889/j.issn.1002-1302.2024.20.015
摘要:為在自然環(huán)境下準確地識別和檢測香梨果實,以YOLO v7為基礎(chǔ)網(wǎng)絡(luò)模型,針對果園中香梨果實、果葉、枝干之間相互遮擋問題,提出一種改進的輕量化YOLO v7梨果目標檢測方法。該方法將MobileNet v3引入YOLO v7 模型中作為其骨干特征提取網(wǎng)絡(luò),從而減少網(wǎng)絡(luò)的參數(shù)量,使其更容易部署在移動端和生產(chǎn)實際,在特征融合層引入?yún)f(xié)同注意力機制CA(coordinate attention)模塊,以提高網(wǎng)絡(luò)的特征表達能力,將原YOLO v7中的損失函數(shù)CIoU替換為SIoU,從而提高模型的檢測速度和性能。(剩余11199字)