TensorFlow JS - Load A Model Generated Using Python

Question

I followed the steps in this tutorial to convert a trained TensorFlow model generated using Python. Now I want to use that to re-create the model in TensorFlow JS. I passed the path for model.js like this:

const model = tf.loadLayersModel(pathToModelsDotJSON).then(() => {
    // Do stuff
})

I'm using Vue.js and when I run the code it throws this error:

Uncaught (in promise) Error: The first layer in a Sequential model must get an `inputShape` or `batchInputShape` argument.
    at new t (tf-layers.esm.js?271e:17)
    at t.add (tf-layers.esm.js?271e:17)
    at t.fromConfig (tf-layers.esm.js?271e:17)
    at deserializeKerasObject (tf-layers.esm.js?271e:17)
    at deserialize (tf-layers.esm.js?271e:17)
    at eval (tf-layers.esm.js?271e:17)
    at eval (tf-layers.esm.js?271e:17)
    at Object.eval [as next] (tf-layers.esm.js?271e:17)
    at o (tf-layers.esm.js?271e:17)

My question is: Why is this happening? And how to solve it?

The error says it all; you need to add batchInputShape to your first model. It might not be necessary in Python but in Js it is — edkeveked
– edkeveked, Commented Jul 14, 2019 at 15:13
@edkeveked And how do I pass it? Should I pass it in the same call to loadLayersModel()? — Hatchi Roku
– Hatchi Roku, Commented Jul 14, 2019 at 15:15
You have to pass it in your python code when you define the model — edkeveked
– edkeveked, Commented Jul 14, 2019 at 15:18
@edkeveked Please post this as an answer so I can accept it. — Hatchi Roku
– Hatchi Roku, Commented Jul 14, 2019 at 20:09

edkeveked · Accepted Answer · 2019-07-15 07:30:45Z

1

1 - You're using the wrong function modelFromJSON to import the model. According to the tutorial, here is the function to use: loadLayersModel.

2-

The first layer in a Sequential model must get an inputShape or batchInputShape argument

You need to add batchInputShape or InputShape to the model defined in Python

edited Jul 15, 2019 at 7:30

answered Jul 14, 2019 at 14:23

edkeveked

18.4k10 gold badges60 silver badges95 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

TensorFlow JS - Load A Model Generated Using Python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related