A peanut and weed detection model used in fields based on BEM-YOLOv7-tiny

Math Biosci Eng. 2023 Oct 17;20(11):19341-19359. doi: 10.3934/mbe.2023855.

Abstract

Due to the different weed characteristics in peanut fields at different weeding periods, there is an urgent need to study a general model of peanut and weed detection and identification applicable to different weeding periods in order to adapt to the development of mechanical intelligent weeding in fields. To this end, we propose a BEM-YOLOv7-tiny target detection model for peanuts and weeds identification and localization at different weeding periods to achieve mechanical intelligent weeding in peanut fields at different weeding periods. The ECA and MHSA modules were used to enhance the extraction of target features and the focus on predicted targets, respectively, the BiFPN module was used to enhance the feature transfer between network layers, and the SIoU loss function was used to increase the convergence speed and efficiency of model training and to improve the detection performance of the model in the field. The experimental results showed that the precision, recall, mAP and F1 values of the BEM-YOLOv7-tiny model were improved by 1.6%, 4.9%, 4.4% and 3.2% for weed targets and 1.0%, 2.4%, 2.2% and 1.7% for all targets compared with the original YOLOv7-tiny. The experimental results of positioning error show that the peanut positioning offset error detected by BEM-YOLOv7-tiny is less than 16 pixels, and the detection speed is 33.8 f/s, which meets the requirements of real-time seedling grass detection and positioning in the field. It provides preliminary technical support for intelligent mechanical weeding in peanut fields at different stages.

Keywords: BEM-YOLOv7-tiny; deep learning; different weeding periods; peanut and weed detection; precision weeding.