Skip to content

A class to interpret CPM caffe models and weights in c++ and a caffe python layer to extract body parts from a person image

Notifications You must be signed in to change notification settings

magomezs/CPMclass

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CPMclass

This repository includes:

  1. A c++ class to interpret CPM caffe deploy (.prototxt) and learnt model (.caffemodel) files, obtaining person skeletons. This class is defined by the files cpm_c++.cpp and cpm_c++.h

  2. A caffe python layer (parts_extractor.py with config file) whose input is a person image blob and its outputs are nine body parts images blobs corresponging to: head, upper right arm, lower right arm, upper laeft arm, lower left arm, upper right leg, lower right leg, upper left leg, and lower left leg. The deploy and learnt model files for the CPM must be given.


Citation:

Please cite CPM c++ class in your publications if it helps your research:

Gómez-Silva, M. J., Armingol, J. M., & de la Escalera, A. (2018). Multi-Object Tracking Errors Minimisation by Visual Similarity and Human Joints Detection. In 8th International Conference on Imaging for Crime Detection and Prevention (ICDP 2017) (pp. 25-30).

@inproceedings{gomez2016multi,
title={Multi-Object Tracking Errors Minimisation by Visual Similarity and Human Joints Detection},
author={G{'o}mez-Silva, Mar{'i}a Jos{'e} and Armingol, Jos{'e} Mar{'i}a and de la Escalera, Arturo},
booktitle={8th International Conference on Imaging for Crime Detection and Prevention (ICDP 2017)},
pages={25--30},
year={2018},
organization={IET}
}


Please cite parts_extractor python layer in your publications if it helps your research:

Gómez-Silva, M. J., Armingol, J. M., & de la Escalera, A. (2018). Deep Parts Similarity Learning for Person Re-Identification. In Proceedings of the 13th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2018) - Volume 5: VISAPP, pages 419-428.

@inproceedings{gomez2017deep,
title={Deep Parts Similarity Learning for Person Re-Identification.},
author={G{'o}mez-Silva, Mar{'\i}a Jos{'e} and Armingol, Jos{'e} Mar{'\i}a and de la Escalera, Arturo},
booktitle={In Proceedings of the 13th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2018)},
pages={419--428},
year={2018}
}

#ACKNOWLEDGMENTS This work was supported by the Spanish Government through the CICYT projects (TRA2015-63708-R and TRA2016-78886-C3-1-R), and Ministerio de Educación, Cultura y Deporte para la Formación de Profesorado Universitario (FPU14/02143), and Comunidad de Madrid through SEGVAUTO-TRIES (S2013/MIT- 2713). We gratefully acknowledge the support of NVIDIA Corporation with the donation of the GPUs used for this research.

About

A class to interpret CPM caffe models and weights in c++ and a caffe python layer to extract body parts from a person image

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published