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

frozen graph or onnx #38

Closed ak9250 closed 4 years ago

ak9250 commented 4 years ago

can the weights be exported to a frozen_graph.pb or model.onnx file?

TachibanaYoshino commented 4 years ago

Sure, this is very simple. I will give the ckpt_to_pb.py and inference_by_pb.py for you.

TachibanaYoshino commented 4 years ago

For onnx file, you can learn it from here

ak9250 commented 4 years ago

@TachibanaYoshino Thanks can you please share the ckpt_to_pb.py and inference_by_pb.py or the frozen_graph.pb file, will it be small enough to fit under 10mb?

TachibanaYoshino commented 4 years ago

What is your email address? I will send you 2 files. But there is a problem with the second file, I believe you can solve it. This is why I did not release them. I don't have time to deal with these problems.

ak9250 commented 4 years ago

@TachibanaYoshino can you send the files to ahsen.khaliq@gmail.com thanks