Keras迁移学习

时间:2020-01-28 10:41:27

标签: python machine-learning keras conv-neural-network transfer-learning

比方说,我想训练一个ConvNet来告诉我照片中是否下雨,程序如何?

鉴于我有两个训练变量trainXtrainYtrainX将是照片,trainY将是标签(例如rainno-rain)。

网络的目标是输出“正确”的答案。问题是:我是否只需要运行model.predict()函数并期待有效的结果?

谢谢您的帮助。

1 个答案:

答案 0 :(得分:2)

1)建立CNN模型:层,激活功能...

2)使用现有的trainXtrainY-数据集对其进行训练。 (最终使用增强来获得更好的结果)

3)使用另一个数据集进行验证,假设它们被称为:testXtestY

4)修改模型的设置,直到精度和损耗足以满足您的需要为止...

5)享受您的CNN

这可以帮助您: Building a CNN with Keras