Optimized DINO model for accurate object detection of sesame seedlings and weeds
摘要
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,