search

jpsml / 6-DOF-Inertial-Odometry

IMU-Based 6-DOF Odometry
BSD 3-Clause "New" or "Revised" License
89 stars 33 forks source link

ValueError: Unknown layer: CustomMultiLossLayer #1

Closed gsh89 closed 4 years ago

gsh89 commented 4 years ago

While running the test.py file with model and the input and ground truth file paths, I am getting the following error

Traceback (most recent call last): File "test.py", line 69, in main() File "test.py", line 24, in main model = load_model(args.model) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/engine/saving.py", line 492, in load_wrapper return load_function(*args, **kwargs) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/engine/saving.py", line 584, in load_model model = _deserialize_model(h5dict, custom_objects, compile) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/engine/saving.py", line 274, in _deserialize_model model = model_from_config(model_config, custom_objects=custom_objects) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/engine/saving.py", line 627, in model_from_config return deserialize(config, custom_objects=custom_objects) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/layers/init.py", line 168, in deserialize printable_module_name='layer') File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/utils/generic_utils.py", line 147, in deserialize_keras_object list(custom_objects.items()))) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/engine/network.py", line 1056, in from_config process_layer(layer_data) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/engine/network.py", line 1042, in process_layer custom_objects=custom_objects) File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/layers/init.py", line 168, in deserialize printable_module_name='layer') File "/Users/Gurtaj/anaconda3/envs/tf/lib/python3.7/site-packages/keras/utils/generic_utils.py", line 140, in deserialize_keras_object ': ' + class_name) ValueError: Unknown layer: CustomMultiLossLayer

Would greatly appreciate your help on this.