Tony607 / object_detection_demo

How to train an object detection model easy for free
https://www.dlology.com/blog/how-to-train-an-object-detection-model-easy-for-free/
Other
177 stars 932 forks source link

Can I convert fine_tuned_models/saved_model.pb to tensorflowjs model ? #8

Open seijichishiki opened 4 years ago

seijichishiki commented 4 years ago

Dear sir,

This repo is very useful for me. Thank you very much.

I want to convert the fine_tuned_models to tensorflowjs model. I tried it, but does not work..

!tensorflowjs_converter --input_format=tf_frozen_model --output_node_names='final_result' --output_format=tensorflowjs --output_json='model.json' saved_model.pb ./web_model/

Using TensorFlow backend. Traceback (most recent call last): File "/usr/local/bin/tensorflowjs_converter", line 8, in sys.exit(main()) File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/converter.py", line 358, in main strip_debug_ops=FLAGS.strip_debug_ops) File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/tf_saved_model_conversion.py", line 356, in convert_tf_frozen_model graph = load_graph(frozen_model_path, output_node_names) File "/usr/local/lib/python3.6/dist-packages/tensorflowjs/converters/tf_saved_model_conversion.py", line 69, in load_graph graph_def.ParseFromString(f.read()) google.protobuf.message.DecodeError: Error parsing message

If you know how to convert, please let me know. Best Regards