加快在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)