search

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

Tensorflow and Tensorflowjs version #34

Open KienPM opened 4 years ago

KienPM commented 4 years ago

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

geni94 commented 4 years ago

@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 commented 4 years ago

@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 commented 4 years ago

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 commented 4 years ago

thank you!