A deep learning approach incorporating YOLO v5 and attention mechanisms for field real-time detection of the invasive weed Solanum rostratum Dunal seedlings

文献类型： 外文期刊

第一作者： Wang, Qifan

作者： Wang, Qifan;Cheng, Man;Cai, Zhenjiang;Yuan, Hongbo;Huang, Shuo;Zhang, Jinlin

作者机构：

关键词： Weed detection; Solanum rostratum Dunal; YOLO v5; Attention mechanism; Multi -scale training; Real-time detection

期刊名称：COMPUTERS AND ELECTRONICS IN AGRICULTURE （ 影响因子：6.757； 五年影响因子：6.817 ）

ISSN： 0168-1699

年卷期： 2022 年 199 卷

页码：

收录情况： SCI

摘要： Solanum rostratum Dunal is a common invasive alien weed that can damage native ecosystems and biodiversity. Detecting Solanum rostratum Dunal at an early stage of growth will make it possible to treat it before it causes serious damage. Therefore, a convolution neural network model YOLO-CBAM is constructed in this paper for the detection of the Solanum rostratum Dunal seedings, which is incorporating YOLO v5 and attention mechanism. A method is designed for slicing the high-resolution images by calculating the overlap rate to construct datasets that reduce the possibility of detail loss due to compressing high-resolution images during the training process. Multiscale training methods have been used to improve training performance. The comparison tests show that the Precision and Recall of the proposed YOLO_CBAM are both higher than that of YOLO v5. The performance of the network is further improved after multi-scale training, and the Average Precision (AP) of YOLO_CBAM increased from 0.9017 to 0.9272. The trained network model was deployed to Jetson AGX Xavier for field trials. The network model achieved a Precision of 0.9465 and a Recall of 0.9017 for real-time recognition. The detection speed and the detection effectiveness can be applied to field real-time detection of the invasive weed Solanum rostratum Dunal seedlings.

分类号：

  • 相关文献
作者其他论文 更多>>

微信小程序