LEHP-DETR: A model with backbone improved and hybrid encoding innovated for flax capsule detection.

Abstract

Flax, as a functional crop with rich essential fatty acids and nutrients, is important in nutrition and industrial applications. However, the current process of flax seed detection relies mainly on manual operation, which is not only inefficient but also prone to error. The development of computer vision and deep learning techniques offers a new way to solve this problem. In this study, based on RT-DETR, we introduced the RepNCSPELAN4 module, ADown module, Context Aggregation module, and TFE module, and designed the HWD-ADown module, HiLo-AIFI module, and DSSFF module, and proposed an improved model, called LEHP-DETR. Experimental results show that LEHP-DETR achieves significant performance improvement on the flax dataset and comprehensively outperforms the comparison model. Compared to the base model, LEHP-DETR reduces the number of parameters by 67.3%, the model size by 66.3%, and the FLOPs by 37.6%. the average detection accuracy mAP50 and mAP50:95 increased by 2.6% and 3.5%, respectively.

Code and data availability

The paper's authors publicly release the LEHP-DETR analysis code (the improved RT-DETR model used for flax capsule detection) on GitHub. The paper-specific FLAX dataset is only available upon request from the lead contact, so it does not qualify as a public asset. DOTA is a cited third-party dataset, not paper-specific