联系平台
  • 国内刊号:CN 32-1115/S
  • 国际刊号:ISSN 0257-4799
  • 主管部门:中国科学技术协会
  • 主办单位:中国蚕学会
    中国农业科学院蚕业研究所
  • 主编:李木旺
  • 地址:江苏省镇江市丹徒区长晖路
    666号 江苏科技大学期刊社
  • 邮政编码:212100
  • 电话:0511-85616835
  • 信箱:canyekexue@126.com
  • Q Q:849229965

基于改进FairMOT模型的多类别蚕茧跟踪算法

【期刊年份】2023年第04期
【作者姓名】孙卫红1,2  于思佳1,2  梁曼1,2  邵铁锋1,2  梅静3
【作者单位】1中国计量大学机电工程学院;2中国计量大学茧丝绸质量检测技术研究所;3江苏省东台市综合检验检测中心

摘要  为获取蚕茧智能分选过程中动态群体蚕茧连续时间的种类及其坐标位置信息,文中提出了一种基于改进FairMOT模型的多类别蚕茧跟踪算法。首先,按类别数扩展预测头的输出维度,并构建多类别损失函数,分别实现对上车茧、黄斑茧和烂茧的跟踪;其次,将原始FairMOT模型中的特征提取网络更改为HarDNet网络,并引入注意力机制,优化蚕茧融合特征;最后,将特征权重同时输入检测和重识别分支进行联合训练,引入GIoU损失函数进行蚕茧检测框尺寸的回归预测,提高检测性能。实验结果表明,文中算法的多目标跟踪准确率达84.6%,整体跟踪帧处理速率为20.9帧/s,多目标跟踪精度、识别F1值、召回率与基于原始FairMOT模型的蚕茧跟踪算法相比分别提高了6.9个百分点、3.6个百分点、2.7个百分点。文中算法可为动态群体蚕茧的分选提供参考。

关键词  注意力机制; GIoU损失函数; 多类别跟踪; 动态分选