一种基于注意力知识蒸馏的轻量级行人重识别方法

    公开(公告)号:CN117351533A

    公开(公告)日:2024-01-05

    申请号:CN202310421455.X

    申请日:2023-04-19

    Applicant: 南通大学

    Inventor: 吕泽 王进 陈海明

    Abstract: 本发明属于计算机视觉技术领域,具体涉及一种基于注意力知识蒸馏的轻量级行人重识别方法。本发明包括以下步骤:步骤1、输入1张彩色图像至轻量级行人重识别特征提取网络;步骤2、利用ResNet50网络和维度交互注意力块,将步骤1中输入的图像生成富含信息的浅层特征和深层特征;步骤3、利用bottleneck,使前三层的特征对齐第四层的特征,获得对齐的浅层特征和深层特征;步骤4、利用深层特征对浅层特征进行注意力迁移知识蒸馏;步骤5、利用步骤3获得的对齐的深层特征对浅层特征进行解耦知识蒸馏;步骤6、若达到指定的训练轮数,则结束;否则继续完成训练,返回步骤1。本发明减少了模型的参数,从而大大提高了模型的运行速度。

    一种基于离线蒸馏的轻量级行人重识别方法

    公开(公告)号:CN117115850A

    公开(公告)日:2023-11-24

    申请号:CN202310562808.8

    申请日:2023-05-18

    Abstract: 本发明公开了一种基于离线蒸馏的轻量级行人重识别方法,在数据集中选取行人的若干张图像构建样本,基于残差网络和归一化的注意力机制,提取样本中图像的特征,通过全连接网络和损失函数,基于提取后的图像特征,训练教师网络,根据获得的教师网络,通过解耦知识蒸馏,协助训练出性能相近的学生网络,并应用该模型,对目标区域的行人进行重识别;本发明提出了基于离线蒸馏的轻量级行人重识别网络,在仅损失少量精度的前提下,极大地降低了网络模型的参数量,有效地提升了行人重识别的推理速度。

Patent Agency Ranking