This project is based on the pytorch-classification.
- prepare dataset and generate gt cvs file and the format is like follows:
path_to_image,pt1_x,pt1_y......
....
- define the model just like in models/basetnet
- set the hyper params in train.py
- take a cup of coffee