human_tracker icon indicating copy to clipboard operation
human_tracker copied to clipboard

human tracking based on yolov3 and center loss

This repository was based on:

  • yolov3 https://github.com/ultralytics/yolov3
  • center-loss https://github.com/louis-she/center-loss.pytorch

performance

  • https://www.bilibili.com/video/av44360925

demo

  • download yolov3 and centerloss weights. https://pan.baidu.com/s/1ECT_CA8UlgsMlzvBcVwr2w pw:om7k
  • python tracking.py the output will be stored in videos/output

train center loss

  • firstly download the training data https://pan.baidu.com/s/1ECT_CA8UlgsMlzvBcVwr2w pw:om7k into train_data/
  • then python main.py