An Improved U-Net Model Based on Multi-Scale Input and Attention Mechanism: Application for Recognition of Chinese Cabbage and Weed

Zhongyang Ma, Gang Wang (), Jurong Yao, Dongyan Huang, Hewen Tan, Honglei Jia and Zhaobo Zou
Additional contact information
Zhongyang Ma: College of Biological and Agricultural Engineering, Jilin University, Changchun 130022, China
Gang Wang: College of Biological and Agricultural Engineering, Jilin University, Changchun 130022, China
Jurong Yao: First Administrative Department of People’s Canal, Sichuan Dujiangyan Water Conservancy Development Center, Chengdu 611000, China
Dongyan Huang: College of Biological and Agricultural Engineering, Jilin University, Changchun 130022, China
Hewen Tan: College of Biological and Agricultural Engineering, Jilin University, Changchun 130022, China
Honglei Jia: College of Biological and Agricultural Engineering, Jilin University, Changchun 130022, China
Zhaobo Zou: Changchun Zhongda Tractor Manufacturing Co., Ltd., Changchun 130062, China

Sustainability, 2023, vol. 15, issue 7, 1-17

Abstract: The accurate spraying of herbicides and intelligent mechanical weeding operations are the main ways to reduce the use of chemical pesticides in fields and achieve sustainable agricultural development, and an important prerequisite for achieving these is to identify field crops and weeds accurately and quickly. To this end, a semantic segmentation model based on an improved U-Net is proposed in this paper to address the issue of efficient and accurate identification of vegetable crops and weeds. First, the simplified visual group geometry 16 (VGG16) network is used as the coding network of the improved model, and then, the input images are continuously and naturally down-sampled using the average pooling layer to create feature maps of various sizes, and these feature maps are laterally integrated from the network into the coding network of the improved model. Then, the number of convolutional layers of the decoding network of the model is cut and the efficient channel attention (ECA) is introduced before the feature fusion of the decoding network, so that the feature maps from the jump connection in the encoding network and the up-sampled feature maps in the decoding network pass through the ECA module together before feature fusion. Finally, the study uses the obtained Chinese cabbage and weed images as a dataset to compare the improved model with the original U-Net model and the current commonly used semantic segmentation models PSPNet and DeepLab V3+. The results show that the mean intersection over union and mean pixel accuracy of the improved model increased in comparison to the original U-Net model by 1.41 and 0.72 percentage points, respectively, to 88.96% and 93.05%, and the processing time of a single image increased by 9.36 percentage points to 64.85 ms. In addition, the improved model in this paper has a more accurate segmentation effect on weeds that are close to and overlap with crops compared to the other three comparison models, which is a necessary condition for accurate spraying and accurate weeding. As a result, the improved model in this paper can offer strong technical support for the development of intelligent spraying robots and intelligent weeding robots.

Keywords: image identification; semantic segmentation; U-Net; ECA; Chinese cabbage; weed (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
https://www.mdpi.com/2071-1050/15/7/5764/pdf (application/pdf)
https://www.mdpi.com/2071-1050/15/7/5764/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jsusta:v:15:y:2023:i:7:p:5764-:d:1107625

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().