2017-10-18 39 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

相關問題