Skip to content

Human body part segmentation model, trained with 22 class labels.

Notifications You must be signed in to change notification settings

Vrushank264/Human-Parsing-PyTorch

Repository files navigation

Human Parsing (Human body Part segmentation) using PyTorch

Results:

Dataset

  • This model is trained on CCIHP dataset which contains 22 class labels.

Please download imagenet pretrained resent-101 from baidu drive or Google drive, and put it into dataset folder.

Training

  • Set necessary arguments and run train_simplified.py.

Citation:

@InProceedings{Liu_2022_CVPR,
    author    = {Liu, Kunliang and Choi, Ouk and Wang, Jianming and Hwang, Wonjun},
    title     = {CDGNet: Class Distribution Guided Network for Human Parsing},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2022},
    pages     = {4473-4482}
}

Releases

No releases published

Packages

No packages published