Awesome
2D-Human-Parsing
Requirements
python == 3.8.8, pytorch == 1.7.1, torchvision == 0.8.2, networkx == 2.6.2
Demo Usage
Download the pretrained model here and put it to the pretrained folder, then simply run:
cd inference
bash demo.sh
Testing on custom data
To test on your own data, just refer to the structure of the demo_imgs folder and the inference/demo.sh file respectively for data preparation and model running. Guess it would be very easy to get start by replacing them with your own data / bash script :).
Parsing Index
Below is the person part index (i.e. pixel value) of the parsing result:
| Part | index | Part | index |
|---|---|---|---|
| background | 0 | neck | 10 |
| hat | 1 | scarf | 11 |
| hair | 2 | skirt | 12 |
| - | 3 | face | 13 |
| sunglass | 4 | left arm | 14 |
| shirt | 5 | right arm | 15 |
| dress | 6 | left leg | 16 |
| coats | 7 | right leg | 17 |
| - | 8 | left shoe | 18 |
| pant | 9 | right shoe | 19 |
Acknowledgement
The code and pretrained model in this repo are provided through the courtesy of Bowen Wu. Thanks for his effort at making this easy-to-use human parsing codebase.
Citation
If you find this repo helpful and use it in your work/paper, please cite:
@misc{BowenParsing2021,
author = {Bowen Wu, Fuwei Zhao},
title = {2D-Human-Parsing},
year = {2021},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/fyviezhao/2D-Human-Parsing}}
}