Interlinked Convolutional Neural Network for Face Parsing

This is a pytorch implementation of the ICNN paper.

How to Run

Copy and unzip Helen dataset in "../data" (one level up from project directory)

Prepare facial parts (Only once) python3 extract_parts.py

Resize dataset (Only once) python3 resize_dataset.py

Create result folder mkdir res/

Train Stage1 python3 train_stage1.py

Train Stage2 python3 train_stage2.py

Modulate background python3 bg_modulate.py

Run end2end python3 end2end.py