Real-Time Detection of Eichhornia crassipes Based on Efficient YOLOV5

被引:6
作者
Qian, Yukun [1 ]
Miao, Yalun [1 ]
Huang, Shuqin [1 ]
Qiao, Xi [2 ]
Wang, Minghui [1 ]
Li, Yanzhou [1 ]
Luo, Liuming [1 ]
Zhao, Xiyong [1 ,2 ]
Cao, Long [1 ]
机构
[1] Guangxi Univ, Sch Mech Engn, Nanning 530004, Peoples R China
[2] Chinese Acad Agr Sci, Agr Genom Inst Shenzhen, Shenzhen Branch,Minist Agr & Rural Affairs, Guangdong Lab Lingnan Modern Agr,Genome Anal Lab, Shenzhen 518120, Peoples R China
关键词
Eichhornia crassipes detection; YOLOV5; deep learning; EfficientNet; GROWTH; MODEL;
D O I
10.3390/machines10090754
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The rapid propagation of Eichhornia crassipes has a threatening impact on the aquatic environment. For most small water areas with good ecology, daily manual monitoring and salvage require considerable financial and material resources. Unmanned boats have important practical significance for the automatic monitoring and cleaning Eichhornia crassipes. To ensure that the target can be accurately detected, we solve the problems that exist in the lightweight model algorithm, such as low accuracy and poor detection effect on targets with small or unclear characteristics. Taking YOLOV5m 6.0 version as the baseline model, given the computational limit of real-time detection, this paper proposes to use EfficientNet-Lite0 as the backbone, use the ELU function as the activation function, modify the pooling mode in SPPF, embed the SA attention mechanism, and add the RFB module in the feature fusion network to improve the feature extraction ability of the whole model. The dataset collected water hyacinth images from ponds and lakes in Guangxi, Yunnan, and the China Plant Image Library. The test results show that efficient YOLOV5 reached 87.6% mAP, which was 7.1% higher than that of YOLOV5s, and the average detection time was 62 FPS. The ablation experiment verifies the effectiveness of each module of efficient YOLOV5, and its detection accuracy and model parameters meet the real-time detection requirements of the Eichhornia crassipes unmanned cleaning boat.
引用
收藏
页数:17
相关论文
共 50 条
[41]   Pavement damage detection model based on improved YOLOv5 [J].
He T. ;
Li H. .
Tumu Gongcheng Xuebao/China Civil Engineering Journal, 2024, 57 (02) :96-106
[42]   A lightweight tea bud detection model based on Yolov5 [J].
Gui, Zhiyong ;
Chen, Jianneng ;
Li, Yang ;
Chen, Zhiwei ;
Wu, Chuanyu ;
Dong, Chunwang .
COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 205
[43]   Ship Target Detection Algorithm Based on Improved YOLOv5 [J].
Zhou, Junchi ;
Jiang, Ping ;
Zou, Airu ;
Chen, Xinglin ;
Hu, Wenwu .
JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2021, 9 (08)
[44]   Mask wearing detection in public places based on YOLOv5 [J].
Liu, Jinqing ;
Zhang, Chao ;
Ma, Boya ;
Wang, Zixu ;
Lei, Shuai .
2021 3RD INTERNATIONAL CONFERENCE ON MACHINE LEARNING, BIG DATA AND BUSINESS INTELLIGENCE (MLBDBI 2021), 2021, :162-165
[45]   Strip Surface Defect Detection Algorithm Based on YOLOv5 [J].
Wang, Han ;
Yang, Xiuding ;
Zhou, Bei ;
Shi, Zhuohao ;
Zhan, Daohua ;
Huang, Renbin ;
Lin, Jian ;
Wu, Zhiheng ;
Long, Danfeng .
MATERIALS, 2023, 16 (07)
[46]   Small gastric polyp detection based on the improved YOLOv5 [J].
Wu, Linfei ;
Liu, Jin ;
Yang, Haima ;
Huang, Bo ;
Liu, Haishan ;
Cheng, Shaowei .
MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (28) :71773-71788
[47]   Surface Defect Detection of Industrial Parts Based on YOLOv5 [J].
Le, Hai Feng ;
Zhang, Lu Jia ;
Liu, Yan Xia .
IEEE ACCESS, 2022, 10 :130784-130794
[48]   Small Object Detection Method based on Improved YOLOv5 [J].
Gao, Tianyu ;
Wushouer, Mairidan ;
Tuerhong, Gulanbaier .
2022 INTERNATIONAL CONFERENCE ON VIRTUAL REALITY, HUMAN-COMPUTER INTERACTION AND ARTIFICIAL INTELLIGENCE, VRHCIAI, 2022, :144-149
[49]   PHOTOVOLTAIC DEFECT DETECTION BASED ON IMPROVED YOLOv5 ALGORITHM [J].
Wang, Yuxin ;
Zhang, Zhi ;
Zhang, Jialiang ;
Han, Jiangning ;
Lian, Jianguo ;
Qi, Yifeng .
Taiyangneng Xuebao/Acta Energiae Solaris Sinica, 2024, 45 (12) :139-145
[50]   A lightweight vehicles detection network model based on YOLOv5 [J].
Dong, Xudong ;
Yan, Shuai ;
Duan, Chaoqun .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2022, 113