来源:期刊VIP网 时间:
作者:张翔凯;李鹏;陈枫赟;余珺泽;于涛;
单位:南京信息工程大学自动化学院;无锡学院自动化学院;
摘要:针对传统的纱线断纱检测方法存在检测精度低、响应速度慢的问题,提出一种基于RT-DETR算法的改进模型TH-DETR。在主干网络采用计算复杂程度低且精度较高的ResNet18,引入OrthoNet正交通道模块改进BasicBlock,通过对多样化通道压缩增强对特征提取。使用具有提取高低频信息的高效注意力HiLo对位置编码器进行改进,更加高效地利用不同频段的信息,提升预测精度,减小冗余。此外,采用Inner-IoU损失函数,通过优化预测框与真实框的内部重叠区域加快模型训练稳定性和收敛速度,提高预测框的准确度。实验结果表明:与基准模型RT-DETR相比,TH-DETR算法平均精度值mAP值达到93.2%,提高了2.0个百分点,计算量降低0.5 G,检测速率达到了83.33帧/s。该模型能够满足单锭断纱检测的需求。
关键词:纱线;;目标检测;;OrthoNet;;HiLo;;Inner-IoU
基金资助:无锡市社会发展科技示范工程项目(N20191008)