Pest-YOLO: A YOLOv5-Based Lightweight Crop Pest Detection Algorithm
DOI:
https://doi.org/10.46604/ijeti.2024.13748Keywords:
pest detection, YOLOv5n, MobileNetV3, CBAMAbstract
Traditional crop pest detection methods face the challenge of numerous parameters and computations, making it difficult to deploy on embedded devices with limited resources. Consequently, a lightweight network is an effective solution to this issue. Based on you only look once (YOLO)v5, this paper aims to design and validate a lightweight and effective pest detector called pest-YOLO. First, a random background augmentation method is proposed to reduce the prediction error rate. Furthermore, a MobileNetV3-light backbone replaces the YOLOv5n backbone to reduce parameters and computations. Finally, the Convolutional Block Attention Module (CBAM) is integrated into the new network to compensate for the reduction in accuracy. Compared to the YOLOv5n model, the pest-YOLO model’s Parameters and Giga Floating Point Operations (GFLOPs) decrease by about 33% and 52.5% significantly, and the Frames per Second (FPS) increase by approximately 11.1%. In contrast, the Mean Average Precision (mAP50) slightly declines by 2.4%, from 92.7% to 90.3%.
References
J. L. Mu, B. Ma, Y. F. Wang, Z. Ren, S. X. Liu, and J. X. Wang, “Review of Crop Disease and Pest Detection Algorithms Based on Deep Learning,” Transactions of the Chinese Society for Agricultural Machinery, vol. 54, no. s2, pp. 301-313, 2023. (In Chinese)
S. Dong, J. Du, L. Jiao, F. Wang, K. Liu, Y. Teng, et al., “Automatic Crop Pest Detection Oriented Multiscale Feature Fusion Approach,” Insects, vol. 13, no. 6, article no. 554, 2022.
T. Domingues, T. Brandão, and J. C. Ferreira, “Machine Learning for Detection and Prediction of Crop Diseases and Pests: A Comprehensive Survey,” Agriculture, vol. 12, no. 9, article no. 1350, 2022.
P. Kartikeyan and G. Shrivastava, “Review on Emerging Trends in Detection of Plant Diseases Using Image Processing With Machine Learning,” International Journal of Computer Application, vol. 174, no. 11, pp. 39-48, 2021.
C. T. Wang, W. J. Liang, Q. W. Guo, H. Zhong, Y. Gan, and D. Q. Xiao, “Review on Computer-Vision-Based Detection of Agricultural Pests,” Journal of Chinese Agricultural Mechanization, vol. 44, no. 7, pp. 207-213, 2023. (In Chinese)
C. Li, T. Zhen, and Z. Li, “Image Classification of Pests With Residual Neural Network Based on Transfer Learning,” Applied Sciences, vol. 12, no. 9, article no. 4356, 2022.
G. Kang, L. Hou, Z. Zhao, and B. Lang, “Research on the Application of Convolutional Neural Network Based on YOLO Algorithm in Pest Small Target Detection,” 3rd Asia-Pacific Conference on Communications Technology and Computer Science, pp. 131-135, 2023.
W. J. Liang, J. Cao, C. L. Sun, H. X. Cao, and W. Y. Zhang, “Research and Development of Crop Diseases Intelligent Recognition System Based on Deep Learning,” Journal of Chinese Agricultural Mechanization, vol. 44, no. 9, pp. 169-175, 2023. (In Chinese)
P. Ju, Y. Song, Y. J. Zhang, Y. F. Xu, and H. Shao, “Research on Agricultural Crop Diseases and Pests Classification Based on Masked Autoencoding,” Electronic Science and Technology, vol. 37, no. 10, pp. 23-29, 2024. (In Chinese)
F. F. Yang, W. C. Xu, S. D. Chen, and Y. B. Lan, “Rice Disease Image Classification Based on Fusion of Focal Loss and Typical Convolutional Neural Network Structure,” Jiangsu Agricultural Sciences, vol. 51, no. 14, pp. 198-204, 2023. (In Chinese)
X. J. Luo and P. H. Hu, “Research and Implementation of Farm Insect Detection Algorithm Based on Deep Learning,” Journal of East China University of Science and Technology, vol. 50, no. 05, pp. 732-739, 2024. (In Chinese)
J. Wang and C. Xu, “Research on Detection of Navel Orange Pests Based on YOLOv5s,” Industrial Control Computer, vol. 36, no. 07, pp. 105-106+109, 2023. (In Chinese)
Y. M. Liu, K. Hu, J. W. Nie, and T. Q. Xie, “Rice Disease and Pest Identification Based on MSDB-ResNet,” Journal of South China Agricultural University, vol. 44, no. 6, pp. 978-985, 2023. (In Chinese)
Y. He, D. H. Chen, and L. Peng, “Research on Object Detection Algorithm of Economic Forestry Pests Based on Improved YOLOv5,” Journal of Chinese Agricultural Mechanization, vol. 43, no. 04, pp. 106-115, 2022. (In Chinese)
S. Aladhadh, S. Habib, M. Islam, M. Aloraini, M. Aladhadh, and H. S. Al-Rawashdeh, “An Efficient Pest Detection Framework With a Medium-Scale Benchmark to Increase the Agricultural Productivity,” Sensors, vol. 22, no. 24, article no. 9749, 2022.
M. Assiri, E. S. A. Elhameed, A. Kumar, and C. Singla, “Automated Insect Detection and Classification Using Pelican Optimization Algorithm With Deep Learning on Internet of Enabled Agricultural Sector,” SN Computer Science, vol. 5, article no. 576, 2024.
A. Hussain and P. B. Srikaanth, “Leveraging Deep Learning and Farmland Fertility Algorithm for Automated Rice Pest Detection and Classification Model,” KSII Transactions on Internet & Information Systems, vol. 18, no. 4, pp. 959-979, 2024.
A. Tzounis, N. Katsoulas, T. Bartzanas, and C. Kittas, “Internet of Things in Agriculture, Recent Advances and Future Challenges,” Biosystems Engineering, vol. 164, pp. 31-48, 2017.
X. Wu, C. Zhan, Y. K. Lai, M. M. Cheng, and J. Yang, “IP102: A Large-Scale Benchmark Dataset for Insect Pest Recognition,” Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 8787-8796, 2019.
S. Woo, J. Park, J. Y. Lee, and I. S. Kweon, “CBAM: Convolutional Block Attention Module,” Proceedings of the European Conference on Computer Vision, pp. 3-19, 2018.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 Wanbo Luo
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
Copyright Notice
Submission of a manuscript implies: that the work described has not been published before that it is not under consideration for publication elsewhere; that if and when the manuscript is accepted for publication. Authors can retain copyright in their articles with no restrictions. Also, author can post the final, peer-reviewed manuscript version (postprint) to any repository or website.
Since Jan. 01, 2019, IJETI will publish new articles with Creative Commons Attribution Non-Commercial License, under Creative Commons Attribution Non-Commercial 4.0 International (CC BY-NC 4.0) License.
The Creative Commons Attribution Non-Commercial (CC-BY-NC) License permits use, distribution and reproduction in any medium, provided the original work is properly cited and is not used for commercial purposes.
.jpg)
