This project is based on the pytorch-classification.

1. pytorch

1. prepare dataset and generate gt cvs file and the format is like follows:

path_to_image,pt1_x,pt1_y......
....

2. define the model just like in models/basetnet
3. set the hyper params in train.py
4. take a cup of coffee