SieveNet

This is the unofficial implementation of 'SieveNet: A Unified Framework for Robust Image-Based Virtual Try-On'
Paper can be found from here

Dataset downloading and processing

Dataset download instructions and link of dataset can be found from official repo of CP-VTON and VITON
Put dataset in data folder

Usage

Clone the repo and install requirements through pip install -r requirements.txt

Traning

Coarse-to-Fine Warping module

     In config.py set self.datamode='train' and self.stage='GMM'
     then run python train.py
You can observe results while traning in tensorboard as below

Conditional Segmentation Mask generation module

     In config.py set self.datamode='Train' and self.stage='SEG'
     then run python train.py

Segmentation Assisted Texture Translation module

     In config.py set self.datamode='Train' and self.stage='TOM'
     then run python train.py