ANT DETECTION USING YOLOV8: EVALUATION OF DATASET TRANSFER IMPACT
DOI:
https://doi.org/10.17770/etr2024vol2.8040Keywords:
ant, deep learning, pests, precision farmingAbstract
In order to avoid having to fight with aphids and plant virus diseases caused by them in gardens, it is very important to notice ant colonies. As a result we decided to train artificial intelligence to detect ant colonies, then this artificial intelligence can be integrated into an autonomous orchard monitoring system using unmanned aerial vehicles. However, there is restricted availability of open datasets, which contain natural images and region specific species. In the scope of pilot study we decided to train convolutional neural network using ANTS dataset and to test it on small domain-specific dataset to identify the need to collect new dataset. The experiment was completed using the popular architecture YOLOv8. The YOLOv8n and YOLOv8m models trained on ANTS showed accuracy 98% and 99% mAP@0.5. Meanwhile, their accuracy was only 6% and 5% mAP@0.5 respectively testing on our dataset called “WildAnts”. Our pilot study experimentally proves that it is important to collect natural dataset of ant images to train robust artificial intelligence for orchard monitoring using unmanned aerial vehicles. This study will be interesting for all machine learning specialists, because it numerically shows accuracy decrease in the result of dataset transfer.
Downloads
References
G. Benckiser, “Ants and sustainable agriculture. A review,” Agronomy for Sustainable Development, vol. 30, no. 2, pp. 191–199, Apr. 2010, doi: https://doi.org/10.1051/agro/2009026.
M. Wu, X. Cao, M. Yang, X. Cao, and S. Guo, “A dataset of ant colonies’ motion trajectories in indoor and outdoor scenes to study clustering behavior,” GigaScience, vol. 11, Jan. 2022, doi: https://doi.org/10.1093/gigascience/giac096.
C. Abeysinghe, C. Reid, H. Rezatofighi, and B. Meyer, “Tracking Different Ant Species: An Unsupervised Domain Adaptation Framework and a Dataset for Multi-object Tracking.” [Online] Available: https://www.ijcai.org/proceedings/2023/0061.pdf [Accessed: Jan. 25, 2024]
X. Cao, “ANTS--ant detection and tracking,” data.mendeley.com, vol. 3, Oct. 2021, doi: https://doi.org/10.17632/9ws98g4npw.3.
Ultralytics, YOLOv8 GitHub repository. [Online] Available: https://github.com/ultralytics/ultralytics [Accessed: Jan. 15, 2024]
S. Kodors, M. Sondors, G. Lācis, E. Rubauskis, I. Apeināns, and I. Zarembo, “RAPID PROTOTYPING OF PEAR DETECTION NEURAL NETWORK WITH YOLO ARCHITECTURE IN PHOTOGRAPHS,” ENVIRONMENT. TECHNOLOGIES. RESOURCES. Proceedings of the International Scientific and Practical Conference, vol. 1, pp. 81–85, Jun. 2023, doi: https://doi.org/10.17770/etr2023vol1.7293.
R. Bai, F. Shen, M. Wang, J. Lu, and Z. Zhang, “Improving Detection Capabilities of YOLOv8-n for Small Objects in Remote Sensing Imagery: Towards Better Precision with Simplified Model Complexity” Jun. 2023, doi: https://doi.org/10.21203/rs.3.rs-3085871/v1.
A. Bochkovskiy, C.-Y. Wang, and H.-Y. Liao, “YOLOv4: Optimal Speed and Accuracy of Object Detection,” Apr. 2020. Available: https://arxiv.org/pdf/2004.10934.pdf [Accessed: Jan. 12, 2024]
S. Khalid, H. M. Oqaibi, M. Aqib, and Y. Hafeez, “Small Pests Detection in Field Crops Using Deep Learning Object Detection,” Sustainability, vol. 15, no. 8, p. 6815, Jan. 2023, doi: https://doi.org/10.3390/su15086815.
S. Kodors, M. Sondors, I. Apeinans, I. Zarembo, G. Lacis, E. Rubauskis and K. Karklina, “Importance of mosaic augmentation for agricultural image dataset”, 2024.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Ilmars Apeinans, Valdis Tārauds, Lienīte Litavniece, Sergejs Kodors, Imants Zarembo
This work is licensed under a Creative Commons Attribution 4.0 International License.
