所属单位:计算机科学与工程学院
发表刊物:Frontiers in Plant Science
项目来源:自选课题
关键字:object detection;forestry pest detection;DETR;Skip connection;small object detection
摘要: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.
合写作者:贾一新,刘陆洋,党源源,宋诗楠
第一作者:刘冰
论文类型:期刊论文
卷号:14
页面范围:2
ISSN号:1664-462X
是否译文:否
发表时间:2023-09-10
