spheregan-tensorflow's Introduction

SphereGAN-Tensorflow

Simple Tensorflow implementation of SphereGAN (CVPR 2019 Oral) | paper | supplementary materials

Usage

mnist and cifar10 are used inside keras
For your dataset, put images like this:

├── dataset
   └── YOUR_DATASET_NAME
       ├── xxx.jpg (name, format doesn't matter)
       ├── yyy.png
       └── ...

Train

> python main.py --dataset mnist --gan_type sphere --phase train

Test

> python main.py --dataset mnist --gan_type sphere --phase test

Analysis

Inverse of stereographic projection

Moment mode

Results

Score

Image (64x64)

The paper only posted the results for 32x32 images, but I also tried 64x64

Mnist

Cifar10

Lsun bedroom

CelebA

Author

Junho Kim

spheregan-tensorflow's People

Contributors

Stargazers

Watchers

spheregan-tensorflow's Issues

질문이 있어요

안녕하세요 코드보고 공부하고 있는 학생입니다
여쭤보고 싶은게 있어서 질문드립니다

""" Training """
        # divide trainable variables into a group for D and a group for G
        t_vars = tf.trainable_variables()
        d_vars = [var for var in t_vars if 'discriminator' in var.name]
        g_vars = [var for var in t_vars if 'generator' in var.name]

이부분을 각각의 loss 값을 구해 최소화 하면서 weight값을 옵티마이즈 시켜준다고 이해했고 이 vars 의 값들은 weight값이라고 이해했는데

t_vars = tf.trainable_variables()
for var in t_vars: 
     if 'discriminator' in var.name:
           d_vars = [var]
for var in t_vars: 
     if 'generator' in var.name:
           g_vars = [var]

위와 밑의 코드는 같은 코드인가요?
제가 파이썬 문법을 제대로 잘 몰라서 이해가 잘 안되서
일단 밑처럼 이해하고 변환 했는데 제가 이해하고 있는 내용이 맞는지
궁금하여 질문드립니다
감사합니다

Poor Quality?

I have also implemented a SphereGAN, but the quality is not as good as the paper claimed. The results of your code seem also not good enough. Do you have any idea?

Recommend Projects