Keras预训练模型跟踪

Question

运行这一行时：

base_model = MobileNetV2(include_top=False, weights='imagenet', input_tensor=input_tensor, pooling=None, classes=NB_CLASS)

我得到以下错误

我该如何解决这个问题？

Answer 1

如果代码中没有保存加载文件的部分，那么很难从哪里判断问题所在。如果我完全理解的话，你在一个文件h5py中保存了权重？

我认为保存整个经过训练的模型总是比只保存权值更容易：

from keras.models import load_model
model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

Answer 2

通过删除预先训练的权重并再次运行代码来重新下载文件，设法解决了这个问题。

预先训练的权重文件保存在/home中，保存在/.keras隐藏文件夹中的.h5文件中。