Berk Ali Çam
Berk Ali Çam

Reputation: 11

How to make Tensorflowjs converter work for .pb files?

I am trying to convert my .pb file (Frozen model format) to TensorFlow.js graph model format. I followed the instructions in the tfjs_converter documentation. So the command that I am running on my terminal is;

tensorflowjs_converter --input_format=tf_frozen_model --output_node_names='model/Softmax' ./model/moods_mirex-musicnn-msd-1.pb ./modeljs

But I keep getting the error;

KeyError: 'The name "\'model/Softmax\'" refers to an Operation not in the graph.'

To make sure the output_node_names argument that I'm using is correct, I used the netron visualization app. I believe it looks like the output node name should be indeed 'model/Softmax' , so I don't know what's wrong.

It shows here that the last layer node name is model/Softmax)

Am I doing something wrong?

Note: I already made sure that all the paths are correct, and there is no version conflicts.

Upvotes: 1

Views: 37

Answers (0)

Related Questions