Tensorflow and Tensorflowjs version about arbitrary-image-stylization-tfjs

[geni94]

@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?

[KienPM]

@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'"

[geni94]

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.

[KienPM]

thank you!