Optimized DINO model for accurate object detection of sesame seedlings and weeds

用于精确检测芝麻幼苗和杂草的优化DINO模型

阅读:2

Abstract

The application of intelligent agricultural machinery is crucial in modern agricultural production. However, in environments where the target and the surrounding morphology are highly similar, such as distinguishing sesame seedlings from weeds, the problem essentially becomes one of optimizing edge detection algorithms for similar targets. To address this issue in agricultural object detection, we developed a custom dataset containing 1,300 images of sesame seedlings and weeds. To overcome the high complexity and low detection accuracy limitations of the original DINO model for this problem, the backbone network was replaced with MobileNet V3, the SENet attention mechanism and neck structure were optimized, and the H-Swish6 activation function was introduced to suit edge devices. Given the higher degree of lignification in the stems of sesame seedlings, these modifications improved the overall Average Precision (AP) of the model on the COCO dataset by 5.1% compared to the original DINO model. Specifically, [Formula: see text] and [Formula: see text] increased by 3.3% and 3.8%, respectively, while [Formula: see text] and [Formula: see text] increased by 2.3% and 3.2%. The model's parameter count was reduced to 29M, inference time was lowered by 60%, and computational cost in FLOPs decreased by 43.72%. To verify the effectiveness of the improvements, we developed a custom dataset containing 1,300 images of sesame seedlings and weeds. On this model, the improved DINO model achieved a maximum AP of 81.8%, outperforming the YOLOv7 model by 5.6%, with an FPS of 24 frames per second. Ablation experiments verified the effectiveness of the model improvements.However, the aforementioned studies have not addressed the issue of low detection accuracy in scenarios with similar targets in the agricultural domain.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。