2017-10-18 29 views
0

训练数据从两个.npy文件中读取。说,train_set被认为是X,train_label被认为是Y.因此,它不是一个多输入的情况。我的任务需要以不同方式增加图像补丁。那么如何为不同的补丁定义不同的图像生成器?虽然有可能是一个很大的补丁,我使用3个补丁作为一个例子:如何将不同的图像生成器应用于Keras中的不同批次?

为PATCH1:

datagen = ImageDataGenerator(rotation_range = 20)

为PATCH2:

datagen = ImageDataGenerator (rotation_range = 40)

为patch3:

datagen = ImageDataGenerator(rotation_range = 60)

如何在不同的补丁上应用不同的生成器,以及如何在描述的场景中使用model.fit(...)或model.fit_generator(...)?

此外,有没有一种方法来旋转图像的特定程度,而不是范围?

谢谢!

回答

0

我自己并没有这样做,但我认为一种方法是使用第一个数据并将第一组训练数据与fit_generator和选定数目的时期一起传递。然后,保存体重并使用fit_generator使用第二个datagen和第二个组。您还需要设置initial_epoch,并且还需要加载权重。为了概括这个问题,你需要做的是用第二个数据库继续训练。请参阅https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model

+0

谢谢迈赫迪。这是否意味着即使只有一个特定的时代,我仍然需要以接力的方式进行训练?还需要继续加载当前批次培训的以前批次权重? – xiaoxiang

+0

在应用第二个数据记录之前,您需要加载权重。 – Mehdi

相关问题