foamliu/Look-Into-Person-PyTorch

Human Parsing with DeepLabv3 in PyTorch.

Look Into Person

Human Parsing with DeepLabv3 in PyTorch.

Dependencies

• NumPy
• PyTorch
• OpenCV

Dataset

Follow the instruction to download Look-Into-Person dataset.

Architecture

DeepLab-v3+

Usage

Data Pre-processing

Extract training images:

$ python extract.py

Train

$ python train.py

If you want to visualize during training, run in your terminal:

$ tensorboard --logdir runs

Demo

Download pre-trained model and put it into models folder.

$ python demo.py

Input	Merged	Output