4

我在尝试保存 Keras 模型时遇到了一些麻烦:

这是我的代码:

import h5py
from keras.models import load_model

try:
    import h5py
    print ('import fine')
except ImportError:
    h5py = None

left.save('left.h5')  # creates a HDF5 file 'my_model.h5'
left_load = load_model('left.h5')

但即使代码打印,我也收到以下错误'import fine'

import fine
---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-145-b641e79036fa> in <module>()
      8     h5py = None
      9 
---> 10 left.save('left.h5')  # creates a HDF5 file 'my_model.h5'

/usr/local/lib/python3.4/dist-packages/keras/engine/topology.py in save(self, filepath, overwrite, include_optimizer)
   2504         """
   2505         from ..models import save_model
-> 2506         save_model(self, filepath, overwrite, include_optimizer)
   2507 
   2508     def save_weights(self, filepath, overwrite=True):

/usr/local/lib/python3.4/dist-packages/keras/models.py in save_model(model, filepath, overwrite, include_optimizer)
     53 
     54     if h5py is None:
---> 55         raise ImportError('`save_model` requires h5py.')
     56 
     57     def get_json_type(obj):

ImportError: `save_model` requires h5py.
4

2 回答 2

5

确保您使用最新版本的 Keras。

此外,这个错误已在 keras github 中报告: https ://github.com/fchollet/keras/issues/3426

在 Linux 上:

sudo apt-get install libhdf5
sudo pip install h5py
于 2017-11-21T21:38:00.293 回答
0

你试过直接安装h5py吗?http://docs.h5py.org/en/latest/build.html

尝试运行 pip install h5py

于 2017-12-29T18:18:27.380 回答