A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps
In a greenhouse environment where tomatoes are grown, pests in yellow sticky traps need to be detected in order to control the pest population. However, tomato pests typically found on yellow sticky traps are small in size and lack distinct visual features, making it difficult for convolutional netw...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2024-12-01
|
| Series: | Agronomy |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2073-4395/14/12/2887 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1846106378193600512 |
|---|---|
| author | Song Wang Daqing Chen Jianxia Xiang Cong Zhang |
| author_facet | Song Wang Daqing Chen Jianxia Xiang Cong Zhang |
| author_sort | Song Wang |
| collection | DOAJ |
| description | In a greenhouse environment where tomatoes are grown, pests in yellow sticky traps need to be detected in order to control the pest population. However, tomato pests typically found on yellow sticky traps are small in size and lack distinct visual features, making it difficult for convolutional networks to extract sufficient contextual information, thereby rendering the tasks of localization and classification exceptionally challenging. In this work, an improved approach based on the advanced object detection model You Only Look Once version 7-tiny (YOLOv7-tiny) is introduced, aiming to enhance the accuracy of detecting small tomato pests while maintaining computational complexity. Firstly, a context information extraction block (CIE) based on a Transformer encoder is proposed, and this block aims to capture global context, explore potential relationships between features, and emphasize important characteristics. Secondly, an Tiny-ELAN fusion network is introduced, which enhanced the feature fusion ability of the network. Thirdly, the feature fusion part takes the P2 feature layer into account and adds a P2 small target detection head. Finally, the SCYLLA-IoU (SIoU) loss function is introduced, and its components are redefined to incorporate direction information, which enhances the model’s learning ability and convergence performance. Experimental results show that our method can accurately detect three insects: whitefly (WF), macrolophus (MR), and nesidiocoris (NC) in the yellow sticky trap images of tomato crops. Compared with Faster R-CNN, SSD, YOLOv3-tiny, YOLOv5s, YOLOv7-tiny, YOLOv7, YOLOv7-x, YOLOv8n, YOLOv8s, YOLOv10n, and RT-DETR, the mean average precision of our method increased by 3.14%, 11.8%, 4.7%, 4.7%, 4.4%, 3.5%, 2.9%, 4.6%, 4.4%, 4.2%, and 4.2%, respectively. |
| format | Article |
| id | doaj-art-dce42eba5d4e4b63b7cf1c4046d2f826 |
| institution | Kabale University |
| issn | 2073-4395 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Agronomy |
| spelling | doaj-art-dce42eba5d4e4b63b7cf1c4046d2f8262024-12-27T14:04:17ZengMDPI AGAgronomy2073-43952024-12-011412288710.3390/agronomy14122887A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect TrapsSong Wang0Daqing Chen1Jianxia Xiang2Cong Zhang3School of Mathematics and Computer, Wuhan Polytechnic University, Wuhan 430048, ChinaSchool of Mathematics and Computer, Wuhan Polytechnic University, Wuhan 430048, ChinaSchool of Mathematics and Computer, Wuhan Polytechnic University, Wuhan 430048, ChinaSchool of Electrical and Electronic Engineering, Wuhan Polytechnic University, Wuhan 430023, ChinaIn a greenhouse environment where tomatoes are grown, pests in yellow sticky traps need to be detected in order to control the pest population. However, tomato pests typically found on yellow sticky traps are small in size and lack distinct visual features, making it difficult for convolutional networks to extract sufficient contextual information, thereby rendering the tasks of localization and classification exceptionally challenging. In this work, an improved approach based on the advanced object detection model You Only Look Once version 7-tiny (YOLOv7-tiny) is introduced, aiming to enhance the accuracy of detecting small tomato pests while maintaining computational complexity. Firstly, a context information extraction block (CIE) based on a Transformer encoder is proposed, and this block aims to capture global context, explore potential relationships between features, and emphasize important characteristics. Secondly, an Tiny-ELAN fusion network is introduced, which enhanced the feature fusion ability of the network. Thirdly, the feature fusion part takes the P2 feature layer into account and adds a P2 small target detection head. Finally, the SCYLLA-IoU (SIoU) loss function is introduced, and its components are redefined to incorporate direction information, which enhances the model’s learning ability and convergence performance. Experimental results show that our method can accurately detect three insects: whitefly (WF), macrolophus (MR), and nesidiocoris (NC) in the yellow sticky trap images of tomato crops. Compared with Faster R-CNN, SSD, YOLOv3-tiny, YOLOv5s, YOLOv7-tiny, YOLOv7, YOLOv7-x, YOLOv8n, YOLOv8s, YOLOv10n, and RT-DETR, the mean average precision of our method increased by 3.14%, 11.8%, 4.7%, 4.7%, 4.4%, 3.5%, 2.9%, 4.6%, 4.4%, 4.2%, and 4.2%, respectively.https://www.mdpi.com/2073-4395/14/12/2887pest detectionYOLOv7Transformer encodercomputer visiondeep learning |
| spellingShingle | Song Wang Daqing Chen Jianxia Xiang Cong Zhang A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps Agronomy pest detection YOLOv7 Transformer encoder computer vision deep learning |
| title | A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps |
| title_full | A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps |
| title_fullStr | A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps |
| title_full_unstemmed | A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps |
| title_short | A Deep-Learning-Based Detection Method for Small Target Tomato Pests in Insect Traps |
| title_sort | deep learning based detection method for small target tomato pests in insect traps |
| topic | pest detection YOLOv7 Transformer encoder computer vision deep learning |
| url | https://www.mdpi.com/2073-4395/14/12/2887 |
| work_keys_str_mv | AT songwang adeeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT daqingchen adeeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT jianxiaxiang adeeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT congzhang adeeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT songwang deeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT daqingchen deeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT jianxiaxiang deeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps AT congzhang deeplearningbaseddetectionmethodforsmalltargettomatopestsininsecttraps |