TP-YOLO: A Lightweight Attention-Based Architecture for Tiny Pest Detection

  • Yang Di*
  • , Son Lam Phung*
  • , Julian Van Den Berg
  • , Jason Clissold
  • , Abdesselam Bouzerdoum*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Automatic detection of agricultural pests is a challenging problem that is of great interest in biosecurity and precision agriculture. The detection model must cope well with the dense distribution of small-sized pests in complex backgrounds. This paper proposes a lightweight attention-based network, called TP-YOLO, for tiny pest detection. We introduce two attention-based components, namely Contextual Transformer and Omni-Dimensional Dynamic Convolution modules, to enhance feature extraction. The proposed modules are integrated into the YOLOv8 backbone, a state-of-the-art baseline for object detection. This paper also introduces a new benchmark dataset consisting of 1,600 images of Khapra beetles for objective evaluation of pest detection algorithms. Extensive experiments on two datasets indicate that TP-YOLO achieves competitive detection accuracy while having a significantly smaller model size and fast prediction time. We have made the code available to the public at: https://github.com/yangdi-cv/TP-YOLO.

Original languageEnglish
Title of host publication2023 Ieee International Conference On Image Processing, Icip
PublisherIEEE Computer Society
Pages3394-3398
Number of pages5
ISBN (Electronic)9781728198354
DOIs
Publication statusPublished - 2023
Event30th IEEE International Conference on Image Processing, ICIP 2023 - Kuala Lumpur, Malaysia
Duration: 8 Oct 202311 Oct 2023

Publication series

NameIeee International Conference On Image Processing Icip

Conference

Conference30th IEEE International Conference on Image Processing, ICIP 2023
Country/TerritoryMalaysia
CityKuala Lumpur
Period8/10/2311/10/23

Keywords

  • Attention mechanism
  • Cnn
  • Pest detection
  • Vision transformers
  • Yolo

Fingerprint

Dive into the research topics of 'TP-YOLO: A Lightweight Attention-Based Architecture for Tiny Pest Detection'. Together they form a unique fingerprint.

Cite this