https://github.com/morvanzhou/celeba-stylegan
https://github.com/morvanzhou/celeba-stylegan
Last synced: 10 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/morvanzhou/celeba-stylegan
- Owner: MorvanZhou
- Created: 2020-12-05T23:58:40.000Z (over 5 years ago)
- Default Branch: main
- Last Pushed: 2021-01-25T08:58:55.000Z (over 5 years ago)
- Last Synced: 2025-04-10T14:28:06.338Z (about 1 year ago)
- Language: Python
- Size: 4.97 MB
- Stars: 12
- Watchers: 2
- Forks: 1
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
An implementation of StyleGAN on CelebA dataset.
## Install
```shell script
git clone https://github.com/MorvanZhou/celebA-styleGAN
cd celebA-styleGAN
pip install -r requirements.txt
```
## Process data
Download CelebA
- [img_align_celeba.zip](https://drive.google.com/file/d/0B7EVK8r0v71pZjFTYXZWM3FlRnM/view?usp=sharing) (~1.4GB)
- [list_attr_celeba.txt](https://drive.google.com/file/d/0B7EVK8r0v71pblRyaVFSWGxPY0U/view?usp=sharing) (25MB)
to celebA_img_align folder, then unzip img_align_celeba.zip to the same folder.
process data to tensorflow tensor_record format.
```shell script
python dataset.py --data_dir ~/data/celebA_img_align/
```
## Train
```shell script
python train.py --data_dir ~/data/celebA_img_align/ -b 32 -w 2 --soft_gpu --latent 128 --lambda_ 10 --epoch 151 -lr 0.0005 -b1 0. -b2 0.99
```
## Results
First few of epoch:
After one day:
