Agronomy (Sep 2024)

Performance Analysis of YOLO and Detectron2 Models for Detecting Corn and Soybean Pests Employing Customized Dataset

  • Guilherme Pires Silva de Almeida,
  • Leonardo Nazário Silva dos Santos,
  • Leandro Rodrigues da Silva Souza,
  • Pablo da Costa Gontijo,
  • Ruy de Oliveira,
  • Matheus Cândido Teixeira,
  • Mario De Oliveira,
  • Marconi Batista Teixeira,
  • Heyde Francielle do Carmo França

DOI
https://doi.org/10.3390/agronomy14102194
Journal volume & issue
Vol. 14, no. 10
p. 2194

Abstract

Read online

One of the most challenging aspects of agricultural pest control is accurate detection of insects in crops. Inadequate control measures for insect pests can seriously impact the production of corn and soybean plantations. In recent years, artificial intelligence (AI) algorithms have been extensively used for detecting insect pests in the field. In this line of research, this paper introduces a method to detect four key insect species that are predominant in Brazilian agriculture. Our model relies on computer vision techniques, including You Only Look Once (YOLO) and Detectron2, and adapts them to lightweight formats—TensorFlow Lite (TFLite) and Open Neural Network Exchange (ONNX)—for resource-constrained devices. Our method leverages two datasets: a comprehensive one and a smaller sample for comparison purposes. With this setup, the authors aimed at using these two datasets to evaluate the performance of the computer vision models and subsequently convert the best-performing models into TFLite and ONNX formats, facilitating their deployment on edge devices. The results are promising. Even in the worst-case scenario, where the ONNX model with the reduced dataset was compared to the YOLOv9-gelan model with the full dataset, the precision reached 87.3%, and the accuracy achieved was 95.0%.

Keywords