Qidian213 / WatchNET

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Pytorch0.4.1_WatchNet: Efficient and Depth-based Network for People Detection in Video Surveillance Systems

Test using the trained model

python pose_detect.py models/posenet.pth
 

Train your model

This is a training procedure using https://www.idiap.ch/dataset/unicity dataset.

Train with unicity dataset

For each 1000 iterations, the recent weight parameters are saved as a weight file model_iter_1000.

python train.py

More configuration about training are in the entity.py file

About


Languages

Language:Python 100.0%