keras_to_tensorflow icon indicating copy to clipboard operation
keras_to_tensorflow copied to clipboard

Published 20 hours ago • verified publisher icon amir-abdi

Unable to convert .h5 file to .pb file, unknown layer BatchNormalizationV1

Open Rajkiran93 opened this issue 5 years ago • 1 comments

I got this error while converting .h5 file to .pb file, but when BatchNormalizationV1 is removed while training the .h5 file is converted to .pb file successfully.

Keras version 2.2.4

Traceback (most recent call last): File "keras_to_tensorflow.py", line 165, in app.run(main) File "/anaconda3/lib/python3.7/site-packages/absl/app.py", line 300, in run _run_main(main, args) File "/anaconda3/lib/python3.7/site-packages/absl/app.py", line 251, in _run_main sys.exit(main(argv)) File "keras_to_tensorflow.py", line 111, in main model = load_model(FLAGS.input_model, FLAGS.input_model_json) File "keras_to_tensorflow.py", line 89, in load_model raise wrong_file_err File "keras_to_tensorflow.py", line 60, in load_model model = tf.keras.models.load_model(input_model_path) File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/engine/saving.py", line 230, in load_model model = model_from_config(model_config, custom_objects=custom_objects) File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/engine/saving.py", line 310, in model_from_config return deserialize(config, custom_objects=custom_objects) File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/layers/serialization.py", line 64, in deserialize printable_module_name='layer') File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/utils/generic_utils.py", line 173, in deserialize_keras_object list(custom_objects.items()))) File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/engine/sequential.py", line 339, in from_config custom_objects=custom_objects) File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/layers/serialization.py", line 64, in deserialize printable_module_name='layer') File "/anaconda3/lib/python3.7/site-packages/tensorflow/python/keras/utils/generic_utils.py", line 163, in deserialize_keras_object raise ValueError('Unknown ' + printable_module_name + ': ' + class_name) ValueError: Unknown layer: BatchNormalizationV1

Rajkiran93 avatar Mar 06 '19 19:03 Rajkiran93

Never mind , got it fixed by installing "tf-nightly" Command:- "pip install tf-nightly"

Rajkiran93 avatar Mar 07 '19 20:03 Rajkiran93