StyleGAN from scratch (pytorch)
{: width="60%" height="70%"}
- 학습 잘 하는 줄 알았는데 inference 결과가 에러... (fix 예정)
├── StyleGAN
├── datasets
│
├── data
│ ├── __init__.py
│ └── dataset.py
├── option.py
├── model.py
├── train.py
├── inference.py
└── README.mddata/__init__.py: dataset loaderdata/dataset.py: data preprocess & get itemmodel.py: Define block and construct Modeloption.py: Environment setting
# Clone this repo and install dependency
git clone https://github.com/inhopp/StyleGAN.gitpython3 train.py
--device {}(defautl: cpu) \
--lr {}(default: 0.0002) \
--n_epoch {}(default: 30) \
--num_workers {}(default: 4) \python3 inference.py
--device {}(defautl: cpu) \
--num_workers {}(default: 4) \https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master