StarGAN v2-Tensorflow - Simple Tensorflow implementation of StarGAN v2

Overview

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

You might also like...
Implementation of Restricted Boltzmann Machine (RBM) and its variants in Tensorflow

xRBM Library Implementation of Restricted Boltzmann Machine (RBM) and its variants in Tensorflow Installation Using pip: pip install xrbm Examples Tut

Functional TensorFlow Implementation of Singular Value Decomposition for paper Fast Graph Learning

tf-fsvd TensorFlow Implementation of Functional Singular Value Decomposition for paper Fast Graph Learning with Unique Optimal Solutions Cite If you f

StyleGAN2 - Official TensorFlow Implementation
StyleGAN2 - Official TensorFlow Implementation

StyleGAN2 - Official TensorFlow Implementation

An efficient and effective learning to rank algorithm by mining information across ranking candidates. This repository contains the tensorflow implementation of SERank model. The code is developed based on TF-Ranking.
An efficient and effective learning to rank algorithm by mining information across ranking candidates. This repository contains the tensorflow implementation of SERank model. The code is developed based on TF-Ranking.

SERank An efficient and effective learning to rank algorithm by mining information across ranking candidates. This repository contains the tensorflow

Implementation of Perceiver, General Perception with Iterative Attention in TensorFlow
Implementation of Perceiver, General Perception with Iterative Attention in TensorFlow

Perceiver This Python package implements Perceiver: General Perception with Iterative Attention by Andrew Jaegle in TensorFlow. This model builds on t

Minimal implementation of Denoised Smoothing: A Provable Defense for Pretrained Classifiers in TensorFlow.
Minimal implementation of Denoised Smoothing: A Provable Defense for Pretrained Classifiers in TensorFlow.

Denoised-Smoothing-TF Minimal implementation of Denoised Smoothing: A Provable Defense for Pretrained Classifiers in TensorFlow. Denoised Smoothing is

Unofficial Implementation of MLP-Mixer in TensorFlow
Unofficial Implementation of MLP-Mixer in TensorFlow

mlp-mixer-tf Unofficial Implementation of MLP-Mixer [abs, pdf] in TensorFlow. Note: This project may have some bugs in it. I'm still learning how to i

Tensorflow implementation for Self-supervised Graph Learning for Recommendation

If the compilation is successful, the evaluator of cpp implementation will be called automatically. Otherwise, the evaluator of python implementation will be called.

Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.

PAWS-TF 🐾 Implementation of Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting View Assignments with Support Samples (PAWS)

Comments
  • what is your training parameters?

    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 ??

    opened by Johnson-yue 2
  • refer test results are blurry

    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 !!!

    opened by sunkyya 3
  • bad training results : large green area on face

    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?

    opened by sunkyya 12
  • Distributed training with TensorFlow

    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.

    opened by phongnhhn92 0
Owner
Junho Kim
Research Scientist
Junho Kim
TensorFlow implementation of "A Simple Baseline for Bayesian Uncertainty in Deep Learning"

TensorFlow implementation of "A Simple Baseline for Bayesian Uncertainty in Deep Learning"

YeongHyeon Park 7 Aug 28, 2022
Simple Tensorflow implementation of "Adaptive Convolutions for Structure-Aware Style Transfer" (CVPR 2021)

AdaConv — Simple TensorFlow Implementation [Paper] : Adaptive Convolutions for Structure-Aware Style Transfer (CVPR 2021) Note This repository does no

Junho Kim 26 Nov 18, 2022
Simple Tensorflow implementation of Toward Spatially Unbiased Generative Models (ICCV 2021)

Spatial unbiased GANs — Simple TensorFlow Implementation [Paper] : Toward Spatially Unbiased Generative Models (ICCV 2021) Abstract Recent image gener

Junho Kim 16 Apr 15, 2022
Deploy tensorflow graphs for fast evaluation and export to tensorflow-less environments running numpy.

Deploy tensorflow graphs for fast evaluation and export to tensorflow-less environments running numpy. Now with tensorflow 1.0 support. Evaluation usa

Marcel R. 349 Aug 6, 2022
TensorFlow Ranking is a library for Learning-to-Rank (LTR) techniques on the TensorFlow platform

TensorFlow Ranking is a library for Learning-to-Rank (LTR) techniques on the TensorFlow platform

null 2.6k Jan 4, 2023
Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

Peter Lin 6.5k Jan 4, 2023
Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

Robust Video Matting (RVM) English | 中文 Official repository for the paper Robust High-Resolution Video Matting with Temporal Guidance. RVM is specific

flow-dev 2 Aug 21, 2022
Simple embedding based text classifier inspired by fastText, implemented in tensorflow

FastText in Tensorflow This project is based on the ideas in Facebook's FastText but implemented in Tensorflow. However, it is not an exact replica of

Alan Patterson 306 Dec 2, 2022
Simple helper library to convert a collection of numpy data to tfrecord, and build a tensorflow dataset from the tfrecord.

numpy2tfrecord Simple helper library to convert a collection of numpy data to tfrecord, and build a tensorflow dataset from the tfrecord. Installation

Ryo Yonetani 2 Jan 16, 2022