All Projects → williamyang1991 → TET-GAN

williamyang1991 / TET-GAN

Licence: MIT License
[AAAI 2019] TET-GAN: Text Effects Transfer via Stylization and Destylization

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to TET-GAN

Few Shot Patch Based Training
The official implementation of our SIGGRAPH 2020 paper Interactive Video Stylization Using Few-Shot Patch-Based Training
Stars: ✭ 313 (+322.97%)
Mutual labels:  gan, style-transfer
CariMe-pytorch
Unpaired Caricature Generation with Multiple Exaggerations (TMM 2021)
Stars: ✭ 33 (-55.41%)
Mutual labels:  gan, style-transfer
Hidt
Official repository for the paper "High-Resolution Daytime Translation Without Domain Labels" (CVPR2020, Oral)
Stars: ✭ 513 (+593.24%)
Mutual labels:  gan, style-transfer
CS231n
My solutions for Assignments of CS231n: Convolutional Neural Networks for Visual Recognition
Stars: ✭ 30 (-59.46%)
Mutual labels:  gan, style-transfer
Cartoonify
Deploy and scale serverless machine learning app - in 4 steps.
Stars: ✭ 157 (+112.16%)
Mutual labels:  gan, style-transfer
Zhihu
This repo contains the source code in my personal column (https://zhuanlan.zhihu.com/zhaoyeyu), implemented using Python 3.6. Including Natural Language Processing and Computer Vision projects, such as text generation, machine translation, deep convolution GAN and other actual combat code.
Stars: ✭ 3,307 (+4368.92%)
Mutual labels:  gan, style-transfer
Neural Painters X
Neural Paiters
Stars: ✭ 61 (-17.57%)
Mutual labels:  gan, style-transfer
HistoGAN
Reference code for the paper HistoGAN: Controlling Colors of GAN-Generated and Real Images via Color Histograms (CVPR 2021).
Stars: ✭ 158 (+113.51%)
Mutual labels:  gan, style-transfer
Tsit
[ECCV 2020 Spotlight] A Simple and Versatile Framework for Image-to-Image Translation
Stars: ✭ 141 (+90.54%)
Mutual labels:  gan, style-transfer
Deep Learning With Python
Example projects I completed to understand Deep Learning techniques with Tensorflow. Please note that I do no longer maintain this repository.
Stars: ✭ 134 (+81.08%)
Mutual labels:  gan, style-transfer
Cyclegan Qp
Official PyTorch implementation of "Artist Style Transfer Via Quadratic Potential"
Stars: ✭ 59 (-20.27%)
Mutual labels:  gan, style-transfer
Warpgan
(CVPR 2019 Oral) Style Transfer with Geometric Deformation
Stars: ✭ 215 (+190.54%)
Mutual labels:  gan, style-transfer
Sketch To Art
🖼 Create artwork from your casual sketch with GAN and style transfer
Stars: ✭ 115 (+55.41%)
Mutual labels:  gan, style-transfer
Cyclegan Music Style Transfer
Symbolic Music Genre Transfer with CycleGAN
Stars: ✭ 201 (+171.62%)
Mutual labels:  gan, style-transfer
CoMoGAN
CoMoGAN: continuous model-guided image-to-image translation. CVPR 2021 oral.
Stars: ✭ 139 (+87.84%)
Mutual labels:  gan, style-transfer
PerceptualGAN
Pytorch implementation of Image Manipulation with Perceptual Discriminators paper
Stars: ✭ 119 (+60.81%)
Mutual labels:  gan
Child-Face-Generation
Deep Convolutional Conditional GAN and Supervised CNN for generating children's faces given parents' faces
Stars: ✭ 26 (-64.86%)
Mutual labels:  gan
tf-adain
TensorFlow implementation of the paper "Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization" by Xun Huang and Serge Belongie
Stars: ✭ 61 (-17.57%)
Mutual labels:  style-transfer
wgan-gp
Pytorch implementation of Wasserstein GANs with Gradient Penalty
Stars: ✭ 161 (+117.57%)
Mutual labels:  gan
anime2clothing
Pytorch official implementation of Anime to Real Clothing: Cosplay Costume Generation via Image-to-Image Translation.
Stars: ✭ 65 (-12.16%)
Mutual labels:  gan

TET-GAN

This is a pytorch implementation of the paper.

Shuai Yang, Jiaying Liu, Wenjing Wang and Zongming Guo. TET-GAN: Text Effects Transfer via Stylization and Destylization, Accepted by AAAI Conference on Artificial Intelligence (AAAI), 2019.

[Project] | [Paper] | [Dataset]

It is provided for educational/research purpose only. Please consider citing our paper if you find the software useful for your work.

Usage:

Prerequisites

  • Python 2.7
  • Pytorch 0.4.1
  • matplotlib
  • scipy
  • opencv-python
  • Pillow

Install

  • Clone this repo:
git clone https://github.com/williamyang1991/TET-GAN.git
cd TET-GAN/src

Testing Example

  • Download a pre-trained model from [Google Drive] or [Baidu Cloud] to ../save/
  • Style Transfer with default parameters
    • Results can be found in ../output/

python test.py
  • Destylization with default parameters
python test.py --c2s 0
  • Transfer the style of 26.jpg onto the text image 2.png and save the result as 26_2.png
python test.py --style_name ../data/style/26.png --content_name ../data/content/2.png --name 26_2.png
  • For black and white text images, use option --content_type 1
python test.py --style_name ../data/style/1.png --content_name ../data/content/4.png --content_type 1

Training Examples

Full Training

  • Prepare dataset in ../data/dataset/ where the images are arranged in this way (See [Dataset] for one example)
dataset/style1/train/1.png
dataset/style1/train/2.png
...
dataset/style2/train/1.png
dataset/style2/train/2.png
...
  • The training image consists of a text image and a style image as follows

  • Train TET-GAN with default parameters
python train.py 

Saved model can be found at ../save/

  • Use --help to view more training options
python train.py --help

Oneshot Training

  • Download a pre-trained model from [Google Drive] or [Baidu Cloud] to ../save/

    • Specify the pretrained model to load using the option --load_model_name
  • Finetune TET-GAN on a new style/glyph image pair (supervised oneshot training)

python oneshotfinetune.py --style_name ../data/oneshotstyle/1-train.png

Saved model can be found at ../save/

  • Finetune TET-GAN on a new style image without its glyph counterpart (unsupervised oneshot training)
python oneshotfinetune.py --style_name ../data/oneshotstyle/1-train.png --supervise 0

Saved model can be found at ../save/

  • Use --help to view more finetuning options
python oneshotfinetune.py --help

Contact

Shuai Yang

[email protected]

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].