/uit-shoesgan

Primary LanguageJupyter Notebook

UIT SHOESGAN

Generating shoes designs using StyleGAN2.

Thanks to rosinality for his awesome repository stylegan2-pytorch.

Dataset

The dataset included 70k shoes images that we collected from Zappos.

Installation

pip install -r requirements.txt

Usage

python app.py

Pretrained model

We have trained the 512px model on our datasets (with the left-view) 420k iterations. Due to the limited of computational resources we only used 1 batch size. T

Demo

We have recorded our demo: https://www.youtube.com/watch?v=COl8VeL-DNQ