Skip to content

Convolutional Pose Machines (CPM) in Pytorch

Notifications You must be signed in to change notification settings

zihaog0724/CPM_Pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Convolutional Pose Machines

This is the Pytorch

There are 7 files in this folder

--handpose_data_cpm.py
data loader for Hand Pose dataset

--handpose_no_label.py
data loader for Hand Pose dataset without ground truth

--cpm.py
Pytorch cpm model

--train.py
--test.py --save.py
--predict.py
--conf.text

usage

1 train model

python cpm_train.py   

You may revise the variable in conf.text

train_data_dir =
train_label_dir =
learning_rate = 8e-6
batch_size = 16
epochs = 50
begin_epoch = 0

Thus change the path to your own datasets and train CPM on your own REMEMBER that you may implement new data loader for you own datasets.

After this, you will get models for several epoches. The models are saved in folder ckpt/ like

ckpt/model_epoch10.pth

2 test model

python cpm_Test.py         

After running this, you will get PCK score for each epoch
You can select the best trained models

3 save prediction results

python cpm_save.py    

After step 2, you will know which is the best epoch, thus you should revise conf.text and change the value of best_model

4 apply models on datasets without ground truth

python cpm_predict.py    

This step is for applying trained model on datasets without ground truth

About

Convolutional Pose Machines (CPM) in Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.3%
  • Shell 0.7%