Доброго времени суток, сохранение модели keras идет через метод save() в .h5 файл
При попытке эту модель загрузить через
from tensorflow.keras.models import load_model
model = load_model("model.h5")
Вылетает следующая ошибка:
Traceback (most recent call last):
File "main.py", line 675, in
run.main()
File "main.py", line 671, in recognition
model = load_model("model.h5")
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\saving.py", line 230, in load_model
model = model_from_config(model_config, custom_objects=custom_objects)
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\saving.py", line 310, in model_from_config
return deserialize(config, custom_objects=custom_objects)
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\layers\serialization.py", line 64, in deserialize
printable_module_name='layer')
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\utils\generic_utils.py", line 173, in deserialize_keras_object
list(custom_objects.items())))
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\network.py", line 1292, in from_config
process_layer(layer_data)
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\network.py", line 1278, in process_layer
layer = deserialize_layer(layer_data, custom_objects=custom_objects)
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\layers\serialization.py", line 64, in deserialize
printable_module_name='layer')
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\utils\generic_utils.py", line 173, in deserialize_keras_object
list(custom_objects.items())))
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\layers\core.py", line 800, in from_config
printable_module_name='function in Lambda layer')
File "C:\Users\user\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\utils\generic_utils.py", line 190, in deserialize_keras_object
fn = module_objects.get(function_name)
AttributeError: 'NoneType' object has no attribute 'get'
Но примитивные модели вроде тех, которыми классифицируют mnist сохраняются и загружаются без проблем
Не знаю, что может пригодиться для решения проблемы, если нужно, скину саму модель и функцию с ее компиляцией
Версии:
Python==3.7.0
tensorflow==1.12.0
h5py==2.9.0
Подскажите пожалуйста, в чем может быть проблема?
