Git Product home page Git Product logo

stargan_v2-tensorflow's Introduction

Official Tensorflow implementation Open ! -> Clova AI

StarGAN v2 — Un-official TensorFlow Implementation [Paper] [Pytorch]

: Diverse Image Synthesis for Multiple Domains

Overview

Architecture

Generator Mapping Network Style Encoder & Discriminator

Usage

├── dataset
   └── YOUR_DATASET_NAME
       ├── train
           ├── domain1 (domain folder)
               ├── xxx.jpg (domain1 image)
               ├── yyy.png
               ├── ...
           ├── domain2
               ├── aaa.jpg (domain2 image)
               ├── bbb.png
               ├── ...
           ├── domain3
           ├── ...
       ├── test
           ├── zzz.jpg (any content image)
           ├── www.png
           ├── ...

Train

python main.py --dataset celebA-HQ_gender --phase train --batch_size 2 --gpu_num 4

Test

python main.py --dataset celebA-HQ_gender --phase test

Refer test

python main.py --dataset celebA-HQ_gender --phase refer_test --refer_img_path refer_img.jpg

Our results

Author

Junho Kim

stargan_v2-tensorflow's People

Contributors

taki0112 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

stargan_v2-tensorflow's Issues

refer test results are blurry

Hi, I have question here.
The "test" mode are fine ,the results are nice, but the "refer test" mode results are very blurry.I saved the cycle image(used for cycle loss) and found that these images are blurry too.
I set the cycleweight to 5 but then all the results are bad.
What should I do to improve the quality of the cycle image or the refer test mode results?Can you help me to solve this out?
THANK YOU !!!

Distributed training with TensorFlow

Hi,
Thanks for the code, I am just wondering what happened if I use Distributed training with TensorFlow in your project since I am having 2 GPUs. I see in your code that during the training phase, you split your image data into several GPUs and then feed them in a for loop which you iterate through each GPU.

I am just wondering is this the optimal way to do this since I am not really familiar with Tensorflow so by looking a bit I found this Distributed training with TensorFlow .

Therefore, this is not an issue but I guess your training loop can be improved by applying this.

what is your training parameters?

Hi , I was reimplementing this work but using Pytorch ,and result is very bad.
thank you for your sharing ,and I have some questions :

  1. what is your training parameters , like "gan_type":
    you default gan_type is gan but this type not using gradient panalty ,and author said using R1 GP, so I guess you used gan_type=dragan ? right?

  2. How do you get target_domain label y~ and source_domain label y ?
    in my opinion, the dataloader about this model is more likely classification training.
    data format is [X_img, domain_label ]
    I think domain_label is original domain of x , so it is source_domain label y. But what is target_domain label y~ and how to get it?? just random ??

Style num

Hi, --num_style = 5, which style did you use? Thanks!

Pretrained models

Hi, first of all thanks for your amazing work! Could you provide some pretrained model? Thanks!

bad training results : large green area on face

Hi, thanks for your sharing!
I used your stargan v2 to train celebA, gantype is "hinge",batchsize is "2",sn is "true". I trained about tow days , the training results are bad and having large green area on face. I doubt that should i continue training or change config?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.