Weed Detection in Maize Fields by UAV Images Based on Crop Row Preprocessing and Improved YOLOv4

被引:31
|
作者
Pei, Haotian [1 ,2 ,3 ]
Sun, Youqiang [3 ]
Huang, He [3 ]
Zhang, Wei [1 ,2 ,3 ]
Sheng, Jiajia [3 ,4 ]
Zhang, Zhiying [5 ]
机构
[1] Anhui Univ, Inst Phys Sci, Hefei 230601, Peoples R China
[2] Anhui Univ, Inst Informat Technol, Hefei 230601, Peoples R China
[3] Chinese Acad Sci, Inst Intelligent Machines, Hefei Inst Phys Sci, Hefei 230031, Peoples R China
[4] Univ Sci & Technol China, Sci Isl Branch, Grad Sch, Hefei 230026, Peoples R China
[5] Jiangsu Univ, Inst Sci & Technol Informat, Zhenjiang 212013, Jiangsu, Peoples R China
来源
AGRICULTURE-BASEL | 2022年 / 12卷 / 07期
关键词
crop row mask; weed detection; samples balance; YOLOv4; Meta-ACON; attention module; Soft-NMS; UAV images;
D O I
10.3390/agriculture12070975
中图分类号
S3 [农学(农艺学)];
学科分类号
0901 ;
摘要
Effective maize and weed detection plays an important role in farmland management, which helps to improve yield and save herbicide resources. Due to their convenience and high resolution, Unmanned Aerial Vehicles (UAVs) are widely used in weed detection. However, there are some challenging problems in weed detection: (i) the cost of labeling is high, the image contains many plants, and annotation of the image is time-consuming and labor-intensive; (ii) the number of maize is much larger than the number of weed in the field, and this imbalance of samples leads to decreased recognition accuracy; and (iii) maize and weed have similar colors, textures, and shapes, which are difficult to identify when an UAV flies at a comparatively high altitude. To solve these problems, we propose a new weed detection framework in this paper. First, to balance the samples and reduce the cost of labeling, a lightweight model YOLOv4-Tiny was exploited to detect and mask the maize rows so that it was only necessary to label weeds on the masked image. Second, the improved YOLOv4 was used as a weed detection model. We introduced the Meta-ACON activation function, added the Convolutional Block Attention Module (CBAM), and replaced the Non-Maximum Suppression (NMS) with Soft Non-Maximum Suppression (Soft-NMS). Moreover, the distributions and counts of weeds were analyzed, which was useful for variable herbicide spraying. The results showed that the total number of labels for 1000 images decrease by half, from 33,572 to 17,126. The improved YOLOv4 had a mean average precision (mAP) of 86.89%.
引用
收藏
页数:18
相关论文
共 50 条
  • [1] Improved YOLOv4 for Pedestrian Detection and Counting in UAV Images
    Kong, Hao
    Chen, Zhi
    Yue, Wenjing
    Ni, Kang
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
  • [2] Improved YOLOv4 for Pedestrian Detection and Counting in UAV Images
    Kong, Hao
    Chen, Zhi
    Yue, Wenjing
    Ni, Kang
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
  • [3] Weed Detection in Potato Fields Based on Improved YOLOv4: Optimal Speed and Accuracy of Weed Detection in Potato Fields
    Zhao, Jiawei
    Tian, Guangzhao
    Qiu, Chang
    Gu, Baoxing
    Zheng, Kui
    Liu, Qin
    ELECTRONICS, 2022, 11 (22)
  • [4] UAV Detection Based on Improved YOLOv4 Object Detection Model
    Niu, Run
    Qu, Yi
    Wang, Zhe
    2021 2ND INTERNATIONAL CONFERENCE ON BIG DATA & ARTIFICIAL INTELLIGENCE & SOFTWARE ENGINEERING (ICBASE 2021), 2021, : 25 - 29
  • [5] YOLOv4_Drone: UAV image target detection based on an improved YOLOv4 algorithm
    Tan, Li
    Lv, Xinyue
    Lian, Xiaofeng
    Wang, Ge
    COMPUTERS & ELECTRICAL ENGINEERING, 2021, 93
  • [6] YOLOv4 and Tiny YOLOv4 Based Forage Crop Detection with an Artificial Intelligence Board
    Beyaz, Abdullah
    Gül, Veysel
    BRAZILIAN ARCHIVES OF BIOLOGY AND TECHNOLOGY, 2023, 66
  • [7] Objects Detection of UAV for Anti-UAV Based on YOLOv4
    Shi, Qingbang
    Li, Jun
    PROCEEDINGS OF 2020 IEEE 2ND INTERNATIONAL CONFERENCE ON CIVIL AVIATION SAFETY AND INFORMATION TECHNOLOGY (ICCASIT), 2020, : 1048 - 1052
  • [8] Detection and location of microaneurysms in fundus images based on improved YOLOv4 with IFCM
    Gao, Weiwei
    Fan, Bo
    Fang, Yu
    Shan, Mingtao
    Song, Nan
    IET IMAGE PROCESSING, 2023, 17 (11) : 3349 - 3357
  • [9] Detection of microaneurysms in fundus images based on improved YOLOv4 with SENet embedded
    Gao W.
    Shan M.
    Song N.
    Fan B.
    Fang Y.
    Shengwu Yixue Gongchengxue Zazhi/Journal of Biomedical Engineering, 2022, 39 (04): : 713 - 720
  • [10] Improved YOLOv4 Based on Attention Mechanism for Ship Detection in SAR Images
    Gao, Yunlong
    Wu, Zhiyong
    Ren, Ming
    Wu, Chuan
    IEEE ACCESS, 2022, 10 : 23785 - 23797