Liu Bing, Jia Yixin, Liu Luyang, Dang Yuanyuan, Song Shinan
College of Computer Science and Technology, Jilin University, Changchun, Jilin, China.
School of Computer Science and Engineering, Changchun University of Technology, Changchun, Jilin, China.
Front Plant Sci. 2023 Aug 15;14:1219474. doi: 10.3389/fpls.2023.1219474. eCollection 2023.
Object detection has a wide range of applications in forestry pest control. However, forest pest detection faces the challenges of a lack of datasets and low accuracy of small target detection. DETR is an end-to-end object detection model based on the transformer, which has the advantages of simple structure and easy migration. However, the object query initialization of DETR is random, and random initialization will cause the model convergence to be slow and unstable. At the same time, the correlation between different network layers is not strong, resulting in DETR is not very ideal in small object training, optimization, and performance. In order to alleviate these problems, we propose Skip DETR, which improves the feature fusion between different network layers through skip connection and the introduction of spatial pyramid pooling layers so as to improve the detection results of small objects. We performed experiments on Forestry Pest Datasets, and the experimental results showed significant AP improvements in our method. When the value of IoU is 0.5, our method is 7.7% higher than the baseline and 6.1% higher than the detection result of small objects. Experimental results show that the application of skip connection and spatial pyramid pooling layer in the detection framework can effectively improve the effect of small-sample obiect detection.
目标检测在林业害虫防治中有着广泛的应用。然而,森林害虫检测面临着数据集缺乏和小目标检测准确率低的挑战。DETR是一种基于Transformer的端到端目标检测模型,具有结构简单、易于迁移的优点。然而,DETR的目标查询初始化是随机的,随机初始化会导致模型收敛缓慢且不稳定。同时,不同网络层之间的相关性不强,导致DETR在小目标训练、优化和性能方面不太理想。为了缓解这些问题,我们提出了Skip DETR,它通过跳跃连接和引入空间金字塔池化层来改善不同网络层之间的特征融合,从而提高小目标的检测结果。我们在林业害虫数据集上进行了实验,实验结果表明我们的方法在平均精度(AP)上有显著提高。当交并比(IoU)值为0.5时,我们的方法比基线高7.7%,比小目标检测结果高6.1%。实验结果表明,在检测框架中应用跳跃连接和空间金字塔池化层可以有效提高小样本目标检测的效果。