首页|期刊导航|河北工业科技|基于改进RT-DETR的玉米缺陷粒检测算法

基于改进RT-DETR的玉米缺陷粒检测算法OA

Corn defective kernel detection algorithm based on improved RT-DETR

中文摘要英文摘要

为了解决玉米籽粒缺陷检测中小尺度缺陷特征易丢失、高频细节捕捉不足的问题,提出一种高频缺陷捕捉检测变换器(high-frequency defect capture detection transformer,HDC-DETR)模型.该模型以实时目标检测变换器(real-time detection transformer,RT-DETR)模型为基础,在主干网络中引入高频增强残差块(high-frequency enhanced residual block,HFERB),并结合跨阶段局部网络(cross stage partial network,CSPNet)的设计思想,构建高频捕获卷积网络(high-frequency capture convolutional network,HCCNet);在中间层引入多分支模块(diverse branch block,DBB),构建基于多分支的C3模块(diverse branch block-based C3 module,DBBC3);在特征融合的上采样环节,采用内容感知重组特征(content-aware reassembly of features,CARAFE)算子替代传统上采样方法.在自建的玉米籽粒数据集上,将HDC-DETR模型与当前主流的检测模型进行性能对比;并开展消融实验,验证各改进模块的有效性.结果表明:在自建的玉米籽粒数据集上,HDC-DETR 模型的mAP@50和mAP@50-95分别达到96.48%和68.51%,较原始RT-DETR模型分别提升2.25个百分点和2.23个百分点;该模型的参数量和计算量分别降至14.3 MB和44.7 GFLOPs,较原始模型分别减少28.14%和21.58%.所提模型有效提升了玉米籽粒小尺度缺陷的检测精度,可为农产品质量自动分选系统提供可靠的技术支撑,在粮食收购、深加工前处理等工业化质检场景中具备较大的实际应用价值.

To address the challenges of easily losing small-scale defect features and insufficiently capturing high-frequency details in corn kernel defect detection,a lightweight and high-precision detection model(high-frequency defect capture detection Transformer,HDC-DETR)was proposed.Based on the real-time detection Transformer(RT-DETR),this model integrated the design concept of cross stage partial network(CSPNet)and high-frequency enhanced residual block(HFERB)in the backbone network to construct a high-frequency capture convolutional network(HCCNet)backbone.A diverse branch block(DBB)was introduced into the intermediate layers,based on which a diverse branch block-based C3 module(DBBC3)was constructed.During the upsampling stage of feature fusion,the content-aware reassembly of features(CARAFE)operator was employed to replace the traditional upsampling method.Performance comparisons between HDC-DETR and current mainstream detection models were conducted on a self-built corn kernel dataset,and ablation experiments were carried out to verify the effectiveness of each proposed module.The results demonstrate that on the self-built corn kernel test set,the mAP@50 and mAP@50-95 of HDC-DETR reach 96.48%and 68.51%,respectively,increasing by 2.25 percentage points and 2.23 percentage points compared to the original RT-DETR model.Furthermore,the parameter count and computational complexity of the model are reduced to 14.3 MB and 44.7 GFLOPs,decreasing by 28.14%and 21.58%compared to the baseline model,respectively.The proposed model effectively improves the detection accuracy of small-scale defects in corn kernels,providing reliable technical support for automatic agricultural product quality sorting systems,and holds significant practical application value in industrial quality inspection scenarios such as grain purchasing and preprocessing for deep processing.

王书海;冯亚爽;宿景芳;王震洲;王建超

河北科技大学信息科学与工程学院,河北石家庄 050018河北科技大学信息科学与工程学院,河北石家庄 050018河北科技大学信息科学与工程学院,河北石家庄 050018河北科技大学信息科学与工程学院,河北石家庄 050018河北科技大学信息科学与工程学院,河北石家庄 050018

信息技术与安全科学

计算机图像处理玉米缺陷粒检测小目标检测高频特征提取轻量化模型

computer image processingcorn defective kernel detectionsmall target detectionhigh-frequency feature extractionlightweight model

《河北工业科技》 2026 (3)

226-236,11

河北省高等学校科学技术研究项目(QN2023185)

10.7535/hbgykj.2026yx03004

评论