公开(公告)号：CN118351415A

公开(公告)日：2024-07-16

申请号：CN202410402201.8

申请日：2024-04-03

Applicant: 南京邮电大学

Inventor： 胡海峰 ,  虞钧程

IPC: G06V10/82 ,  G06V10/44 ,  G06V10/52 ,  G06V10/74 ,  G06V10/774 ,  G06V10/80 ,  G06V40/10 ,  G06V20/52 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/0495 ,  G06N3/096 ,  G06N3/084 ,  G06N3/0985

Abstract: 本发明公开了一种基于知识蒸馏的视频场景下行人重识别的轻量化方法，包括如下步骤：S1.下载视频场景下的数据集MARS和iLIDS‑VID，并处理；S2.对教师模型PiT进行训练；S3.构建参数量较小的学生模型TinyViT，并在模型中插入卷积注意力模块MCT；S4.对学生模型进行蒸馏学习。本发明克服了现有模型在行人重识别下参数量较大，训练数据要求高的不足，通过对大模型进行蒸馏学习的方法，在保持较高性能的同时，显著减少了模型的参数量，使得在资源受限的环境下仍具有较好的计算效率。