解决Keras模型加载速度慢
原始模型加载和保存
原始模型保存
model.save('model{0}.h5'.format(n_model))
原始模型加载
models= load_model('model{0}.h5'.format(ix))
解决方法1
改进模型保存
model.save_weights('model.h5')
model_json = model.to_json()
with open('model.json', "w") as json_file:
json_file.write(model_json)
json_file.close()
改进模型加载
from keras.models import model_from_json
json_file = open("model.json", 'r')
loaded_model_json = json_file.read()
json_file.close()
model = model_from_json(loaded_model_json)
model.load_weights("model.h5")
解决方法2
通过在每次加载之前清除keras会话来解决了这个问题
from keras import backend as K
for i in range(...):
K.clear_session()
model = load_model(...)
参考:https://www.it1352.com/1524821.html