python-3.x - 加快在 Keras 中加载模型？

Question

有没有办法加快在 Keras 中加载模型的速度？我正在使用迁移学习的初始模型。根据我的经验，加载模型似乎需要 13 秒。

我想将一些模型加载到智能手机上。使用TensorFlow作为后端。

start = time.time()
path = r'C:\Users\Moondra\Desktop\2017-12-20_10.hdf5'
labels = os.listdir(r'C:\Users\Moondra\Desktop\FISHES_MAIN')
model = load_model(path)

print(time.time() - start)

输出

12.808000087738037

Answer 1

我发现使用 compile=False 调用 load_model 可以加快速度。我的时间平均从 12-15 秒下降到大约 2-3 秒。

load_model(path, compile=False)