基于無人機航拍與改進(jìn)YOLOv5s的油茶果實檢測
中國農(nóng)機化學(xué)報
頁數(shù): 7 2024-08-29
摘要: 針對無人機航拍圖像中油茶果實小且互相遮擋的問題,提出改進(jìn)YOLOv5s模型。首先,使用SPD-Conv代替YOLOv5s模型中池化操作,使模型在執(zhí)行下采樣操作時能夠保留更多細(xì)粒度信息。然后,在YOLOv5s模型的頸部網(wǎng)絡(luò)末端引入坐標(biāo)注意力機制CA,提高模型對遮擋目標(biāo)的魯棒性。另外,改進(jìn)YOLOv5s模型使用NWD邊界框損失函數(shù)替換YOLOv5s中的CIOU邊界框損失函數(shù),以提升... (共7頁)