- 专利标题: 一种基于蒸馏的轻量化无人机单目标跟踪方法
-
申请号: CN202410830906.X申请日: 2024-06-26
-
公开(公告)号: CN118379332B公开(公告)日: 2024-08-16
- 发明人: 王栋 , 张斌 , 陈鑫 , 刘鹏宇
- 申请人: 大连理工大学
- 申请人地址: 辽宁省大连市甘井子区凌工路2号
- 专利权人: 大连理工大学
- 当前专利权人: 大连理工大学
- 当前专利权人地址: 辽宁省大连市甘井子区凌工路2号
- 代理机构: 辽宁鸿文知识产权代理有限公司
- 代理商 李宝元
- 主分类号: G06T7/246
- IPC分类号: G06T7/246 ; G06V20/17 ; G06V10/764 ; G06V10/75 ; G06V10/82 ; G06V20/40 ; G06V10/25
摘要:
本发明提供一种基于蒸馏的轻量化无人机单目标跟踪方法,属于目标跟踪领域。本发明的主干网络为一个轻量化的ViT‑Base模型,其层数和通道数分别为3和128,总参数量仅为1.428M。轻量化后的ViT‑Base模型大大减少了在推理过程中的时间消耗和内存占用,从而达到模型加速和轻量化的目的。为了有效提高模型的性能,本发明采用多层特征蒸馏的方法,将轻量化模型作为学生模型,ViT‑Base大模型作为教师模型完成模型训练。轻量化模型在训练之初加载预训练的MAE权重参数完成模型初始化。加载预训练权重参数,大大提高了模型的收敛速度并有效提高了模型的泛化性和鲁棒性。
公开/授权文献
- CN118379332A 一种基于蒸馏的轻量化无人机单目标跟踪方法 公开/授权日:2024-07-23