reiinakano/arbitrary-image-stylization-tfjs

Tensorflow and Tensorflowjs version

KienPM opened this issue · 4 comments

Hello @reiinakano,
Thanks for your great repository. Can you tell me which version of Tensorflow and Tensorflowjs have you used?
I want to convert Tensorflowjs model to TFLite model but I have some issue.

Look forward to hearing from you soon,
Thanks

@KienPM as I've been working on a fork of this repo, I can answer that:
This uses Tensorflowjs 1.0 (as you can notice in package.json as well "@tensorflow/tfjs": "~1.0.0").

What's the issue you're encountering with TFLite?

@geni94 Thanks for your answer.

I've tried to convert TensorflowJS model to TFlite model according to this answer.

When I run tensorflowjs_converter --input_format"=tfjs_layers_model --output_format=keras tfjs_model.json hdf5_keras_model.hdf5 to convert from TensorflowJS model to Keras model with tensorflow 2.3.0, tensorflowjs 2.3.0, I got "ValueError: Improper config format"

When I run with tensorflow 1.15.0, tensorflowjs 1.7.4, I got "KeyError: 'class_name'"

I literally told you in the first comment that this library uses tensorflow 1.0, and yet you go ahead and try with 1.15, 1.7.4, 2.3.0, etc... The package.json specifies the version: "@tensorflow/tfjs": "~1.0.0". I'm not surprised you can't transform this model with TFLite, seeing that the versions do not match.

thank you!