使用预训练的模型和自己的数据集进行学习转移

时间:2020-01-16 13:52:31

标签: python tensorflow keras

我想训练具有转移学习能力的CNN。数据集是我的,在任何库中都不存在。我看到了本教程(https://www.tensorflow.org/tutorials/images/transfer_learning)。我使用keras.preprocessing加载了图像,如https://www.tensorflow.org/tutorials/load_data/images中所述。我创建了train_data_gen,validation_data_gen和test_data_gen。问题是如何使用这些* _data_gen而不是raw_train,如何使用本教程进行验证和测试。

0 个答案:

没有答案