如何保存经过训练的FCN模型并在经过训练的FCN模型上测试新图像?

时间:2019-08-20 01:02:34

标签: keras

我已经在自己的数据集中训练了FCN,我的代码正在运行。但是,我想保存模型并使用经过训练的模型测试新图像,有人可以帮我吗?

from keras import optimizers


sgd = optimizers.SGD(lr=1E-2, decay=5**(-4), momentum=0.9, nesterov=True)
model.compile(loss='categorical_crossentropy',
          optimizer=sgd,
          metrics=['accuracy'])

hist1 = model.fit(X_train,y_train,
              validation_data=(X_test,y_test),
              batch_size=2,epochs=20,verbose=1)


for key in ['loss', 'acc', 'val_loss', 'val_acc']:
plt.plot(hist1.history[key],label=key)
plt.legend()
plt.show()


y_pred = model.predict(X_test)
y_predi = np.argmax(y_pred, axis=3)
y_testi = np.argmax(y_test, axis=3)
print(y_testi.shape,y_predi.shape)

0 个答案:

没有答案
相关问题