We found a match
Your institution may have access to this item. Find your institution then sign in to continue.
- Title
TIA-YOLOv5: An improved YOLOv5 network for real-time detection of crop and weed in the field.
- Authors
Aichen Wang; Tao Peng; Huadong Cao; Yifei Xu; Xinhua Wei; Bingbo Cui
- Abstract
Introduction: Development of weed and crop detection algorithms provides theoretical support for weed control and becomes an effective tool for the sitespecific weed management. For weed and crop object detection tasks in the field, there is often a large difference between the number of weed and crop, resulting in an unbalanced distribution of samples and further posing difficulties for the detection task. In addition, most developed models tend to miss the small weed objects, leading to unsatisfied detection results. To overcome these issues, we proposed a pixel-level synthesization data augmentation method and a TIA-YOLOv5 network for weed and crop detection in the complex field environment. Methods: The pixel-level synthesization data augmentation method generated synthetic images by pasting weed pixels into original images. In the TIAYOLOv5, a transformer encoder block was added to the backbone to improve the sensitivity of the model to weeds, a channel feature fusion with involution (CFFI) strategy was proposed for channel feature fusion while reducing information loss, and adaptive spatial feature fusion (ASFF) was introduced for feature fusion of different scales in the prediction head. Results: Test results with a publicly available sugarbeet dataset showed that the proposed TIA-YOLOv5 network yielded an F1-scoreweed, APweed and mAP@0.5 of 70.0%, 80.8% and 90.0%, respectively, which was 11.8%, 11.3% and 5.9% higher than the baseline YOLOv5 model. And the detection speed reached 20.8 FPS. Discussion: In this paper, a fast and accurate workflow including a pixel-level synthesization data augmentation method and a TIA-YOLOv5 network was proposed for real-time weed and crop detection in the field. The proposed method improved the detection accuracy and speed, providing very promising detection results.
- Subjects
OBJECT recognition (Computer vision); FIELD crops; WEEDS; WEED control; DATA augmentation; SUGAR beets
- Publication
Frontiers in Plant Science, 2022, Vol 13, p1
- ISSN
1664-462X
- Publication type
Article
- DOI
10.3389/fpls.2022.1091655