本项目目标检测算法使用的是FCHD-Fully-Convolutional-Head-Detector,目标跟踪使用的是在线多实例学习online MIL算法。
-
系统Windows 10,显卡使用的GTX 1060,理论上Linux也可以运行。
-
安装PyTorch >=0.4 with GPU (code are GPU-only)
-
安装 cupy,可以通过
pip install cupy-cuda80
or(cupy-cuda90,cupy-cuda91, etc). -
install visdom for visualization, refer to their github page
- 下载 VGG16 预训练模型,link.下载好后存到
data/pretrained_model
folder. - 下载训练好的模型 link. 放到
checkpoints/
folder. - 下载测试视频video1,video2
- 修改
demo.py
中的视频路径 - 运行
python demo.py