Skip to content

Latest commit

 

History

History
79 lines (53 loc) · 2.65 KB

README.md

File metadata and controls

79 lines (53 loc) · 2.65 KB

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