Rapid Detection and Counting of Wheat Ears in the Field Using YOLOv4 with Attention Module release_odyby3g22jckvehadh6474jqeu

by Baohua Yang, Zhiwei Gao, Yuan Gao, Yue Zhu

Published in Agronomy by MDPI AG.

2021   Volume 11, p1202

Abstract

The detection and counting of wheat ears are very important for crop field management, yield estimation, and phenotypic analysis. Previous studies have shown that most methods for detecting wheat ears were based on shallow features such as color and texture extracted by machine learning methods, which have obtained good results. However, due to the lack of robustness of these features, it was difficult for the above-mentioned methods to meet the detection and counting of wheat ears in natural scenes. Other studies have shown that convolutional neural network (CNN) methods could be used to achieve wheat ear detection and counting. However, the adhesion and occlusion of wheat ears limit the accuracy of detection. Therefore, to improve the accuracy of wheat ear detection and counting in the field, an improved YOLOv4 (you only look once v4) with CBAM (convolutional block attention module) including spatial and channel attention model was proposed that could enhance the feature extraction capabilities of the network by adding receptive field modules. In addition, to improve the generalization ability of the model, not only local wheat data (WD), but also two public data sets (WEDD and GWHDD) were used to construct the training set, the validation set, and the test set. The results showed that the model could effectively overcome the noise in the field environment and realize accurate detection and counting of wheat ears with different density distributions. The average accuracy of wheat ear detection was 94%, 96.04%, and 93.11%. Moreover, the wheat ears were counted on 60 wheat images. The results showed that R2 = 0.8968 for WD, 0.955 for WEDD, and 0.9884 for GWHDD. In short, the CBAM-YOLOv4 model could meet the actual requirements of wheat ear detection and counting, which provided technical support for other high-throughput parameters of the extraction of crops.
In application/xml+jats format

Archived Files and Locations

application/pdf  1.6 MB
file_qejwh7lkwvdcxf6ygdemrckbmi
res.mdpi.com (publisher)
web.archive.org (webarchive)
application/pdf  15.4 MB
file_qywujei6ebac5bjr6uwbhlgwby
mdpi-res.com (web)
web.archive.org (webarchive)
Read Archived PDF
Preserved and Accessible
Type  article-journal
Stage   published
Date   2021-06-12
Language   en ?
Container Metadata
Open Access Publication
In DOAJ
In ISSN ROAD
In Keepers Registry
ISSN-L:  2073-4395
Work Entity
access all versions, variants, and formats of this works (eg, pre-prints)
Catalog Record
Revision: ffb44e3c-b415-452f-8278-1befc829bab1
API URL: JSON