LabelTrack

LabelTrack是为多目标跟踪MOT写的一个自动标注工具

安装

git clone https://github.com/DD-DuDa/LabelTrack.git
cd LabelTrack/Tracking
pip install -r requirements.txt
pip install cython
pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
pip install cython_bbox
python setup.py develop

主要功能

• 导入mp4文件或视频帧文件夹
• 手动标注
• 修改标注框，包括大小，标签，ID等信息
• 采用SOTA目标跟踪模型对视频帧进行预跟踪
• 导出和导入VisDrone格式数据集

快捷键

使用方法

1. 下载对应模型权重
   • bytetrack_m_mot17.pth.tar (code:2zkp)
   • yolox_l_best_ckpt.pth.tar (code:493b)
   • yolox_m_vd.pth.tar (code:y9qv)
   • yolox_tiny_best_ckpt.pth.tar (code:v4u1)
2. 修改./Tracking/configs中的yaml文件（exp_file, ckpt）

   cd LabelTrack
   python main.py

   ./Tracking/videos 有demo视频

已更新

• 【2022.9.20】手动标注框的更新
• 【2022.7.16】上传权重文件（百度网盘）
• 【2022.5.31】取消所有标注框，手动标注框延续所有帧
• 【2022.5.14】标注图片大小缩放，拖拽；工具栏等按钮完善；标注类别有11种

待更新

• [ ] 操作错误提醒完善
• [ ] 手动标注框的ID从没有出现的开始
• [ ] 关于视频帧的加载
• [ ] 更多需要的功能

参考

1. https://github.com/tzutalin/labelImg
2. https://github.com/ifzhang/ByteTrack

B站视频

https://www.bilibili.com/video/BV15Y4y137Fp/