在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

我发现用load_model给compile=False打电话加快了速度。我的时间从12到15秒下降到平均2-3秒。

load_model(path, compile=False)