代码段:
from tensorflow.keras.models import load_model
model_path = 'vgg16_weights_tf_dim_ordering_tf_kernels.h5'
model_vgg = load_model(model_path)
报错信息:
背景知识:
"""
模型保存方式:
save()→保存模型参数和网络结构、损失函数、优化器
save_weights()→仅保存模型参数,可以看作是保存模型的一部分。
"""
#方式1
from tensorflow.keras.models import load_model
model.save('xxx.h5')
model = load_model('xxx.h5')
#方式2
model.save_weights('xxx.h5')
model = model.load_weights('xxx.h5')