保存Keras型号:UTF - 8错误

时间:2018-01-13 04:55:03

标签: utf-8 neural-network hdf5 convolution utf

我在keras中构建了一个卷积神经网络,如下所示:

model = Sequential()

    model.add(Convolution2D(nb_filters, nb_conv, nb_conv,
                            border_mode='valid',
                            input_shape=(img_cols, img_rows, 3)))
    convout1 = Activation('relu')
    model.add(convout1)
    model.add(Convolution2D(nb_filters, nb_conv, nb_conv))
    convout2 = Activation('relu')
    model.add(MaxPooling2D(pool_size=(nb_pool, nb_pool)))
    model.add(convout2)
    model.add(MaxPooling2D(pool_size=(nb_pool, nb_pool)))
    model.add(Dropout(0.5))
    model.add(Flatten())
    model.add(Dense(128))
    model.add(Activation('relu'))
    model.add(Dropout(0.5))
    model.add(Dense(nb_classes))
    model.add(Activation('softmax'))
    model.compile(loss='categorical_crossentropy', optimizer='adadelta', metrics=['accuracy'])

我正在尝试使用以下方法保存模型的权重:

fname = "weights-Test-CNN.hdf5"
model.load_weights(fname)

程序运行,并创建一个文件但是一旦我打开文件,这就是显示的内容:

错误! C://Users/NAME/weights-Test-CNN.hdf5不是UTF-8编码的。 保存已禁用。 有关详细信息,请参阅控制台。

如何修复此错误以便正确保存权重?

1 个答案:

答案 0 :(得分:2)

权重 实际上已保存。这里的问题是您无法将它们作为UTF-8编码文件读取。但是如果你尝试加载权重,它应该可以工作。