search

TachibanaYoshino / AnimeGAN

A Tensorflow implementation of AnimeGAN for fast photo animation ! This is the Open source of the paper 「AnimeGAN: a novel lightweight GAN for photo animation」, which uses the GAN framwork to transform real-world photos into anime images.
4.49k stars 660 forks source link

how to use?i want convert a picture to anime picture. #44

Open wchpeng opened 3 years ago

wchpeng commented 3 years ago

if I want convert a picture to anime picture. What is command should use?

I install environment using a docker image, and put my pictures in folder "samples", and run docker run -it xxxx python main.py

I except AnimeGAN can handle my pictures, but he start train the pictures in folder dataset. one step and one step, slowly, I close it using ctrl+c until it runing step 100+.

I just want convert my pictures to anime picture. What is wrong with me? And what should I do correct？？

TachibanaYoshino commented 3 years ago

First check the gpu settings os.environ["CUDA_VISIBLE_DEVICES"] = "0", Then, check whether the checkpoint weight is stored in the correct directory Finally, in the test.py file, set the default weight path, test image path, output image path, and run.