zaidalyafeai / ml-projects

Implementation of web friendly ML models using TensorFlow.js. pix2pix, face segmentation, fast style transfer and many more ...
MIT License
647 stars 254 forks source link

Pb2json #9

Open MeLonJ10 opened 5 years ago

MeLonJ10 commented 5 years ago

I 've trained the pix2pix model by python, and got a pb or checkpoint file ,but i don't know how to use it in js. Should I transform the pb or ckpt into a json? Can u offer me the .py or any other script to do it. It will be appreciated so much!!!

zaidalyafeai commented 5 years ago

Check this https://github.com/tensorflow/tfjs-converter