Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
dataloader.py		dataloader.py
models.py		models.py
parser.py		parser.py
snapshot_tensorboard_pattern.png		snapshot_tensorboard_pattern.png
test.py		test.py
train.py		train.py
utils.py		utils.py

README.md

Pattern Makeup - Training Guideline


This is a training guideline for Pattern Branch (P), one out of two branches in Color-Pattern Makeup Transfer (CPM)

Requirements: Please refer to Getting Started/ Requirements, the main components are:
- segmentation_models.pytorch
- torch >=1.6

Data Preparation: Please download CPM-Synt-1 in here, or via wget:

wget https://public.vinai.io/CPM-datasets/CPM-Synt-1.zip
unzip CPM-Synt-1.zip

Training/ Evaluation: python train.py --datapath /pathtodata. Test result should be found in Tensorboard.
(Optional):
- Open Tensorboard: tensorboard --logdir=runs
- Change backbones, pretrained_weights: Check parser.py


Snapshot of Tensorboard at epoch 1. Evaluation metrics on test set should be found (Accuracy, Recall, Precision, etc.)