Pest detection model based on multi-scale dataset

被引:0
|
作者
Chen D. [1 ]
Lin J. [1 ]
Wang H. [1 ]
Wu K. [1 ]
Lu Y. [2 ]
Zhou X. [1 ]
Zhang J. [1 ]
机构
[1] School of Automation, Hangzhou Dianzi University, Hangzhou
[2] Hangzhou Ruikun Technology Co., Ltd, Hangzhou
基金
中国国家自然科学基金;
关键词
crop pest; deep learning; model; target detection; transformer;
D O I
10.11975/j.issn.1002-6819.202311113
中图分类号
学科分类号
摘要
Pests have posed the severe threat to the crop yield in agricultural production. The frequent outbreaks of crop pest have limited the development of agriculture in China. It is very necessary to develop the artificial intelligence (AI) monitoring in modern pest control. Among them, transformer-based DETR models have demonstrated the strong potential to the multi-scale tasks of pest detection with intensive distribution that captured from nature scene. But the current quantitative analysis cannot verify the core mechanism of detection models. In this study, the DINO-based model was proposed to detect and count the dense tiny insects. Four modules were contained in the model. Multi-scale deformable attention module (MDAM) was used for the multi-scale feature mapping; Contrastive denoising training strategy (CDTS) was used to alleviate the duplicate detection of single pest object; Mixed query selection and look forward twice (MQS-LFT) was used to obtain the decoder's Query with more efficient initialization and iteration. A total of 4 019 images were collected from the real scene, including planthoppers, aphids and wheat mites. As such, the pest dataset was obtained and then named MTC-PAWPD (multi-scale tiny crowded planthopper-aphid-wheat mite pest detection). The dataset was also divided into the training, validation and test set, according to the ratio of 5:2:3. According to the overlap and relative scale, the dataset was then divided into four scenes using data partitioning, namely, intensive-distribution, large-scale, normal-scale, and tiny-scale scene. A comparison was also made on the performance of model in different scenes. The ablation and validation experiments were carried out to visualize the features and Query Anchor, in order to validate the correlation between the performance and query-related modules. The experimental results demonstrated that the DINO-based model was performed better on the recognition of pest objects. In MTC-PAWPD benchmark, the highest mean average precision was achieved at 50% intersection over union (mAP@50) of 70.0%, indicating an improvement of 2.3 percentage points mAP@50, compared with the mainstream object detection models, such as Faster R-CNN, YOLOv5x, ATSS, YOLOX, and Deformable DETR. In addition, the convergence speed was only 1/10 with respect to Yolo-based model. The mAP@50 reached 42.5%, 79.4%, 75.7% and 62.4% in the four scenes, respectively. All of them showed the powerful detection performance, particularly in the scale-related scenes. All the four improvement modules of DINO were enhanced the performance of pest target detection in real scenes. The performance bias was correlated to the module of Transformer. The performance was substantially improved by 0.2, 1.0, 0.9, and 0.8 percentage point in mAP@50, respectively. Specifically, the mixed query selection strategy was utilized the features in the encoder for query generation. The query point was allowed in the first layer of the decoder closer to the pest objects in the image. Thus, more features were identified to enhance the expressive power of the model. Look forward twice enabled the Query Point in the later layers of the decoder closer to the objects. Contrastive Denoising Training Strategy was introduced into the high-quality negative samples, particularly for the overlapped prediction of a single pest object. In summary, the dense query and components of DINO can be expected to facilitate the pest detection and counting tasks in multiple scale, complex, and densely distributed natural scenarios. The DINO model can fully meet the requirements for the feature extraction from the multi-scale nature pest images, thus accurately detecting the objects. The transformer model can also provide the strong generalization and practical value in the detection of multi-scale pest under the complex background of the field. © 2024 Chinese Society of Agricultural Engineering. All rights reserved.
引用
收藏
页码:196 / 206
页数:10
相关论文
共 42 条
  • [1] JIANG Xinlu, CHEN Tianen, WANG Cong, Et al., Survey of deep learning algorithms for agricultural pest detection, Computer Engineering and Applications, 59, 6, pp. 1-17, (2022)
  • [2] SHEN Yanyan, ZHANG Jingcheng, SHEN Dong, Et al., Research progress on habitat suitability assessment of crop diseases and pests by multi-source remote sensing information, Chinese Journal of Eco-Agriculture, 31, 7, pp. 1012-1025, (2023)
  • [3] YAN Chuan, LI Chunjie, LIN Kejian, Et al., Current situation, research progress and management strategies of plant diseases, insect pests, and rodent pests in grasslands of China, Bulletin of National Natural Science Foundation of China, 37, 4, pp. 580-586, (2023)
  • [4] ZHENG Zengwei, LI Yanzhen, LIU Yi, Et al., Identifying fine-grained vineyard pest using attribute knowledge graph, Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 39, 11, pp. 146-154, (2023)
  • [5] XIE J, ZHANG X, LIU Z, Et al., Detection of litchi leaf diseases and insect pests based on improved FCOS, Agronomy, 13, 5, (2023)
  • [6] YE Yun, ZHAO Xiaojuan, JIANG Sheng, Application and realization of litchi pest identification technology based on deep learning, China Agricultural Informatics, 34, 4, pp. 30-37, (2022)
  • [7] YANG Z, FENG H, RUAN Y, Et al., Tea tree pest detection algorithm based on improved Yolov7-Tiny, Agriculture, 13, 5, (2023)
  • [8] WANG Q, ZHANG S, DONG S, Et al., Pest24: A large-scale very small object data set of agricultural pests for multi-target detection, Computers and Electronics in Agriculture, 175, (2020)
  • [9] LIU L, WANG R, XIE C, Et al., PestNet: An end-to-end deep learning approach for large-scale multi-class pest detection and classification, IEEE Access, 7, pp. 45301-45312, (2019)
  • [10] JIAO L, XIE C, CHEN P, Et al., Adaptive feature fusion pyramid network for multi-classes agricultural pest detection, Computers and Electronics in Agriculture, 195, (2022)