search

pvitoria / ChromaGAN

Official Implementation of ChromaGAN: An Adversarial Approach for Picture Colorization
GNU Affero General Public License v3.0
198 stars 38 forks source link

How to convert my_model_colorization.h5 to tensorfow lite model to cut model size? #6

Closed rollmark closed 4 years ago

rollmark commented 4 years ago

Hi, thanks for your great work, current pretraind model almost 800M, but my computer only 1G memory to use, how can i convert this keras model to tensorflow lite model, then i can load this model to memory to have fun, Thank you!

pvitoria commented 4 years ago

Hi @rollmark, Sorry for the late reply. We have no experience in tensorflow lite model. So we can not help there. If you get to any solution let us know :) Best, Patricia