International Journal of Advances in Intelligent Informatics
Vol 9, No 3 (2023): November 2023

Deep learning pest detection on Indonesian red chili pepper plant based on fine-tuned YOLOv5

Indra Agustian (Study Program of Electrical Engineering, University of Bengkulu)
Ruvita Faurina (Study Program of Informatics, University of Bengkulu)
Sahrial Ihsani Ishak (Study Program of Informatics, University of Bengkulu)
Ferzha Putra Utama (Study Program of Informatics, University of Bengkulu)
Kusmea Dinata Dinata (Agricultural Technology Study Center, Agriculture Department, Province of Bengkulu)
Novalio Daratha (Study Program of Electrical Engineering)



Article Info

Publish Date
01 Nov 2023

Abstract

.This research developed a pest detection model for Indonesian red chili pepper based on fine-tuned YOLOv5. Indonesian red chili pepper is the third largest vegetable commodity produced in Indonesia. Pest attacks disrupt the quantity and quality of crop yields. To control pests effectively, it is necessary to detect the type of pest correctly. A viable solution is to leverage computer vision and deep learning technologies. However, no previous studies have developed a pest detection model for Indonesian red chili pepper based on this technology. YOLOv5 is a variant of the YOLO object detection algorithm, which has major advantages in terms of computation cost and execution speed. The dataset comprises 4,994 image files collected from a chili plantation in Bengkulu province, Indonesia, covering 4 different classes and a total of 10,683 pests. The image is 1216 x1216 px with the smallest, largest, and average object dimensions of 2%, 35%, and 4% of the image dimensions. The training model used is fine-tuning YOLOv5s with variations of patience as an early stop parameter of 100, 200, and 300. The evaluation of the trained model is based on train loss, validation loss, and mAP@0.5:0.95, the best-trained model is the 445th epoch on patience 100 with the best confidence value of 0.321 and the highest TF1 of 0.74. From the best-trained model testing on the test dataset, the mAP@0.5 performance for all classes is 81.3%. The model not only detected large pests but was also able to detect objects that were small in size compared to the image size. The best-trained model's best mAP@0.5 performance and speed are 82.6% and 20 ms/image, or 50 fps on NVIDIA P100 GPU.

Copyrights © 2023






Journal Info

Abbrev

IJAIN

Publisher

Subject

Computer Science & IT

Description

International journal of advances in intelligent informatics (IJAIN) e-ISSN: 2442-6571 is a peer reviewed open-access journal published three times a year in English-language, provides scientists and engineers throughout the world for the exchange and dissemination of theoretical and ...