Part Grouping Network (PGN)
Ke Gong, Xiaodan Liang, Yicheng Li, Yimin Chen, Ming Yang and Liang Lin, "Instance-level Human Parsing via Part Grouping Network", ECCV 2018 (Oral).
Introduction
PGN is a state-of-art deep learning methord for semantic part segmentation, instance-aware edge detection and instance-level human parsing built on top of Tensorflow.
This distribution provides a publicly available implementation for the key model ingredients reported in our latest paper which is accepted by ECCV 2018.
Crowd Instance-level Human Parsing (CIHP) Dataset
The PGN is trained and evaluated on our CIHP dataset for isntance-level human parsing. Please check it for more model details. The dataset is also available at google drive and baidu drive.
Pre-trained models
We have released our trained models of PGN on CIHP dataset at google drive.
Inference
- Download the pre-trained model and store in $HOME/checkpoint.
- Prepare the images and store in $HOME/datasets.
- Run test_pgn.py.
- The results are saved in $HOME/output
Training
- Download the pre-trained model and store in $HOME/checkpoint.
- Download CIHP dataset or prepare your own data and store in $HOME/datasets.
- For CIHP dataset, you need to generate the edge labels and left-right flipping labels (optional). We have provided a script for reference.
- Run train_pgn.py to train PGN.
- Use test_pgn.py to generate the results with the trained models.