/LabelTrack-Tool

LabelTrack是一个针对于多目标跟踪的图形化自动标注平台

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

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格式数据集

快捷键

键位 功能
w 标注
s, del 删除所选标注框
ctrl + 滚轮 放大,缩小

使用方法

  1. 下载对应模型权重
  1. 修改./Tracking/configs中的yaml文件(exp_file, ckpt)
cd LabelTrack
python main.py

./Tracking/videos 有demo视频 3. handle change annos.txt result cmd:

3.1 delete error influence-IDs:
sed -i -e '/,1|/d' annos.txt
3.2 replace target-IDs:(48 --> 1)
sed -i 's/,48|/,1|/' annos.txt

已更新

  • 【2022.9.20】手动标注框的更新

  • 【2022.7.16】上传权重文件(百度网盘)

  • 【2022.5.31】取消所有标注框,手动标注框延续所有帧

  • 【2022.5.14】标注图片大小缩放,拖拽;工具栏等按钮完善;标注类别有11种

  • 【2024.4.26】target_id using the "|" apart from boxes[x,y,w,h]

  • 待更新

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

参考

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

B站视频

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