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

Abstract

Summary: 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.

Keywords

• Bioinformatics
• Plant biology
• Agricultural science