All Projects → VITA-Group → Transgan

VITA-Group / Transgan

Licence: other
[Preprint] "TransGAN: Two Transformers Can Make One Strong GAN", Yifan Jiang, Shiyu Chang, Zhangyang Wang

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Transgan

Nn
🧑‍🏫 50! Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
Stars: ✭ 5,720 (+562.04%)
Mutual labels:  gan, transformer
Unsupnts
Unsupervised Neural Text Simplification
Stars: ✭ 23 (-97.34%)
Mutual labels:  gan
Anime Inpainting
An application tool of edge-connect, which can do anime inpainting and drawing. 动漫人物图片自动修复,去马赛克,填补,去瑕疵
Stars: ✭ 761 (-11.92%)
Mutual labels:  gan
Lightning Bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
Stars: ✭ 829 (-4.05%)
Mutual labels:  gan
Pytorch Pretrained Biggan
🦋A PyTorch implementation of BigGAN with pretrained weights and conversion scripts.
Stars: ✭ 779 (-9.84%)
Mutual labels:  gan
Began Tensorflow
Tensorflow implementation of "BEGAN: Boundary Equilibrium Generative Adversarial Networks"
Stars: ✭ 904 (+4.63%)
Mutual labels:  gan
Getting Things Done With Pytorch
Jupyter Notebook tutorials on solving real-world problems with Machine Learning & Deep Learning using PyTorch. Topics: Face detection with Detectron 2, Time Series anomaly detection with LSTM Autoencoders, Object Detection with YOLO v5, Build your first Neural Network, Time Series forecasting for Coronavirus daily cases, Sentiment Analysis with BERT.
Stars: ✭ 738 (-14.58%)
Mutual labels:  transformer
Inverse Style Gan
Looking up a generative latent vectors from (face) reference images.
Stars: ✭ 26 (-96.99%)
Mutual labels:  gan
Advanced Deep Learning With Keras
Advanced Deep Learning with Keras, published by Packt
Stars: ✭ 917 (+6.13%)
Mutual labels:  gan
Turbotransformers
a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.
Stars: ✭ 826 (-4.4%)
Mutual labels:  transformer
Tensorlayer
Deep Learning and Reinforcement Learning Library for Scientists and Engineers 🔥
Stars: ✭ 6,796 (+686.57%)
Mutual labels:  gan
Musegan
An AI for Music Generation
Stars: ✭ 794 (-8.1%)
Mutual labels:  gan
Odsc 2020 nlp
Repository for ODSC talk related to Deep Learning NLP
Stars: ✭ 20 (-97.69%)
Mutual labels:  transformer
Instagan
InstaGAN: Instance-aware Image Translation (ICLR 2019)
Stars: ✭ 761 (-11.92%)
Mutual labels:  gan
Nag
[CVPR 2018] Tensorflow implementation of NAG : Network for Adversary Generation
Stars: ✭ 24 (-97.22%)
Mutual labels:  gan
Gans In Action
Companion repository to GANs in Action: Deep learning with Generative Adversarial Networks
Stars: ✭ 748 (-13.43%)
Mutual labels:  gan
Bert Keras
Keras implementation of BERT with pre-trained weights
Stars: ✭ 820 (-5.09%)
Mutual labels:  transformer
Xdf Gan
A GAN for the generation of mock astronomical surveys
Stars: ✭ 17 (-98.03%)
Mutual labels:  gan
Figma Transformer
A tiny utility library that makes the Figma API more human friendly.
Stars: ✭ 27 (-96.87%)
Mutual labels:  transformer
Cell Detr
Official and maintained implementation of the paper Attention-Based Transformers for Instance Segmentation of Cells in Microstructures [BIBM 2020].
Stars: ✭ 26 (-96.99%)
Mutual labels:  transformer

TransGAN: Two Transformers Can Make One Strong GAN

Code used for TransGAN: Two Transformers Can Make One Strong GAN.

News

checkpoint for generating images on celeba64 dataset is released!

Main Pipeline

Main Pipeline

Visual Results

Visual Results

prepare fid statistic file

mkdir fid_stat

Download the pre-calculated statistics (Google Drive) to ./fid_stat.

Environment

pip install -r requirements.txt

Notice: Pytorch version has to be <=1.3.0 !

Training

Coming soon

Testing

Firstly download the checkpoint from (Google Drive) to ./pretrained_weight

# cifar-10
sh exps/cifar10_test.sh

# stl-10
sh exps/stl10_test.sh

# celeba64
sh exps/celeba64_test.sh

Acknowledgement

Codebase from AutoGAN, pytorch-image-models

Citation

if you find this repo is helpful, please cite

@article{jiang2021transgan,
  title={TransGAN: Two Transformers Can Make One Strong GAN},
  author={Jiang, Yifan and Chang, Shiyu and Wang, Zhangyang},
  journal={arXiv preprint arXiv:2102.07074},
  year={2021}
}
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].