人体检测 算法 python_多人跟踪（基于centerNet的人体检测器+带有pytorch的深度排序算法）...

centerNet + deep sort with pytorch

This is multi-people tracking code ( centerNet[1] version of yolov + deepsort[2] ), which implemented on CUDA 9.0, ubuntu 16.04, and Anaconda python 3.6. We used CenterNet for real-time object tracking.

Install

conda env create -f CenterNet.yml

pip install -r requirments.txt

Quick Start

Change CENTERNET_ROOT to your local directory in demo_centernet_deepsort.py.

CENTERNET_PATH = 'CENTERNET_ROOT/CenterNet/src/lib/'

to

e.g) CENTERNET_PATH = '/home/kyy/centerNet-deep-sort/CenterNet/src/lib/'

Run demo

Using sample video, we can track multi person.

python demo_centernet_deepsort.py

for webcam, modify two lines

opt.input_type = 'webcam'

//webcam device number

opt.webcam_ind = 0

for ip camera, modify three lines

opt.i