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

摘要： 为获取蚕茧智能分选过程中动态群体蚕茧连续时间的种类及其坐标位置信息，文中提出了一种基于改进FairMOT模型的多类别蚕茧跟踪算法。首先，按类别数扩展预测头的输出维度，并构建多类别损失函数，分别实现对上车茧、黄斑茧和烂茧的跟踪；其次，将原始FairMOT模型中的特征提取网络更改为HarDNet网络，并引入注意力机制，优化蚕茧融合特征；最后，将特征权重同时输入检测和重识别分支进行联...