融合多分支与多粒度特征的东北虎重识别
Re-Identification of Amur Tigers by Combining Multi-branch and Multi-granularity Features作者机构:东北林业大学信息与计算机工程学院哈尔滨150040 哈尔滨学院哈尔滨150086 黑龙江林业智能装备工程研究中心哈尔滨150040 呼伦贝尔学院内蒙古呼伦贝尔021008
出 版 物:《哈尔滨理工大学学报》 (Journal of Harbin University of Science and Technology)
年 卷 期:2023年第28卷第6期
页 面:95-102页
学科分类:0810[工学-信息与通信工程] 08[工学] 081002[工学-信号与信息处理]
基 金:黑龙江省自然科学基金项目(LH2020F040) 中央高校基本科研业务费专项资金资助项目(No.2572017PZ10)
主 题:东北虎重识别 残差网络 多分支特征 坐标注意力机制 circle loss
摘 要:为解决东北虎重识别研究中存在的细节特征提取不充分等问题,提出了一种融合多分支与多粒度特征的东北虎重识别模型CMM-Net。其中,全局分支负责提取宏观上的粗粒度特征;注意力分支通过插入坐标注意力模块加深了网络对重要特征的关注度;局部分支通过将特征图切分成不同条带块,从而提取东北虎更细粒度的局部特征。通过多个分支结构和多个细粒度特征结合来对模型进行优化学习,加强全局特征与局部特征的关联性。同时提出用Circle Loss与Softmax的联合损失来提高网络精度。实验结果表明,在ATRW数据集上所提模型在单摄像头环境下mAP为93.6%,跨摄像头环境下mAP为77.4%,均优于多数文献所提方法,证明了本文模型的有效性。