Direct Export to tensorflow js

lobe / web-bootstrap

Bootstrap your Lobe machine learning model with our web starter project.

https://lobe.ai

MIT License

73 stars 32 forks source link

Direct Export to tensorflow js #14

Closed KartoffelToby closed 3 years ago

KartoffelToby commented 3 years ago

a lobe export Option to tensorflow js would be nice.

adammenges commented 3 years ago

Agreed! We'd love to add this and are evaluating when we could right now.

KartoffelToby commented 3 years ago

for now a working workflow to convert it would be nice. but this is related to #13

kmanadkat commented 3 years ago

I just tried to convert one & miserably failed. I did get model.json but the web code crashed badly.

I followed tfjs-converter as suggested in Docs. I assumed that the output of Lobe Export is Tensorflow SavedModel
I got the model.json file (along with several bin files) with many warnings in terminal & there was no metadata.json file.

Please have a tensorflow js as the export option. Anyone can help for now?

PS : I am just trying to create a simple Male / Female classifier for experimenting. It works like charm on Lobe application, but unable to bring it to web.

ohjs88 commented 3 years ago

I don't understand why they made an example in a format that they couldn't extract.

jlarmstrongiv commented 3 years ago

After installing and using tensorflowjs_wizard, the converter throws an error with AssertionError: Got '4194304'.

Has anyone had any luck converting to a tensorflow.js? Any workaround, no matter how hacky, would be super helpful.

mbeissinger commented 3 years ago

We will have Lobe directly export TensorFlow.js soon! For now, you can use the conversion script here to go from the TensorFlow saved model exported by Lobe into TensorFlow.js: https://github.com/mbeissinger/tfjs-converter