中文

LightTrack:通过一次性架构搜索为目标跟踪寻找轻量神经网络

计算机视觉与模式识别 2021-04-30 v1

摘要

目标跟踪在过去几年取得了显著进展。然而,最先进的跟踪器变得越来越繁重和昂贵,限制了其在资源受限应用中的部署。在这项工作中,我们提出LightTrack,它使用神经架构搜索(NAS)来设计更轻量和高效的目标跟踪器。综合实验表明我们的LightTrack是有效的。它能找到比手工设计的SOTA跟踪器(如SiamRPN++和Ocean)性能更优、同时模型Flops和参数量少得多的跟踪器。此外,当部署在资源受限的移动芯片上时,所发现的跟踪器运行速度快得多。例如,在Snapdragon 845 Adreno GPU上,LightTrack比Ocean快12×12\times,同时参数量少13×13\times、Flops少38×38\times。此类改进可能缩小目标跟踪任务中学术模型与工业部署之间的差距。LightTrack发布于 https://github.com/researchmm/LightTrack。

关键词

引用

@article{arxiv.2104.14545,
  title  = {LightTrack: Finding Lightweight Neural Networks for Object Tracking via One-Shot Architecture Search},
  author = {Bin Yan and Houwen Peng and Kan Wu and Dong Wang and Jianlong Fu and Huchuan Lu},
  journal= {arXiv preprint arXiv:2104.14545},
  year   = {2021}
}

备注

Accepted by CVPR 2021, Github: https://github.com/researchmm/LightTrack