31sy / PGECNet

PGECNet

This respository includes a PyTorch implementation of PGECNet.

Requirements:

python 3.6
PyTorch 0.4.1

Compiling

InPlace-ABN have a native CUDA implementation, which must be compiled with the following commands:

cd modules
sh build.sh
python build.py

Note the CUDA kernels need to update accoding to your own gpu.

Dataset and pretrained model

Plesae download LIP dataset

pretrained model(MM:ydtv)

imagenet pretrained resnett101(MM:ydtv)

Evaluation

bash job_evaluate_val.sh

Training

bash job_train_ge_multi.sh

Acknowledgment

This project is created based on the CE2P.

If this code is helpful for your research, please cite the following paper:

@article{PGEC2019,
title={Human Parsing with Pyramidical Gather-Excite Context},
author={Sanyi Zhang, Guo-jun Qi, Xiaochun Cao, Zhanjie Song, Jie Zhou},
journal={IEEE Transactions on Circuits and Systems for Video Technology(TCSVT)},
year={2021},
volume={31},
number={3},
pages={1016-1030}
}