reiinakano / arbitrary-image-stylization-tfjs

Arbitrary style transfer using TensorFlow.js
https://reiinakano.github.io/arbitrary-image-stylization-tfjs/
Apache License 2.0
1.19k stars 213 forks source link

How about migrate the model to the mobile? #6

Closed carachan98 closed 5 years ago

carachan98 commented 5 years ago

Great demo for arbitrary style transfer! The progress from fixed style images to totally free choices is really exciting. It seems that you are more likely to create a demo with js? Inspired by this repo, I want to have a try with migrating the model to the mobile like Prisma, just for practice, no commercial use : ). Since I am not familiar with the tensorflow.js apis, there are few questions:

Is there any way to integrate the .js with android frame? Or the possible solution is taking use of the original model (before combined with js) and trying to reduce it with apis like tensorflow lite?

Looking for your help : )

reiinakano commented 5 years ago

Hi! You can check #22. It seems @khanhlvg will work on the same thing. Perhaps he could give you some tips since I have never done anything with TF Lite