Skip to content

Tensorflow implementation of GAN and white-box representations of images by observing the behavior of cartoon paintings and consulting artists.

Notifications You must be signed in to change notification settings

tuanPhantom/White-Box-anime-style-GAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Apply an animated style to Images using GAN and White-box Representation

Use cases

Scenery

Food

Indoor Scenes

People

Prerequisites

  • Training code: Linux or Windows
  • NVIDIA GPU + CUDA CuDNN for performance
  • Inference code: Linux, Windows and MacOS

How To Use

Installation

  • Assume you already have NVIDIA GPU and CUDA CuDNN installed
  • Install tensorflow-gpu
  • Install scikit-image==0.14.5, other versions may cause problems
  • Install numpy==1.23.1, other versions may cause problems

Inference with Pre-trained Model

  • Store test images in /test_code/test_images
  • Run /test_code/cartoonize.py
  • Results will be saved in /test_code/cartoonized_images

Train

  • Place your training data in corresponding folders in /dataset
  • Run pretrain.py, results will be saved in /pretrain folder
  • Run train.py, results will be saved in /train_cartoon folder
  • Codes are cleaned from production environment and untested
  • There may be minor problems but should be easy to resolve
  • Pretrained VGG_19 model can be found at following url: https://drive.google.com/file/d/1s7rLnubm25IpZEGUwRmR75kbTCUo3GfI/view?usp=share_link

Datasets

License

About

Tensorflow implementation of GAN and white-box representations of images by observing the behavior of cartoon paintings and consulting artists.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages