【发布时间】:2017-12-23 20:28:40
【问题描述】:
有没有办法加快在 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
【问题讨论】:
-
您使用的是哪个后端?
-
@MatiasValdenegro Tensorflow
标签: python-3.x deep-learning keras keras-2