Проблема с использованием keras ImageDataGenerator для вывода нескольких изображений - PullRequest
0 голосов
/ 12 февраля 2020

Я работаю над использованием keras ImageDataGenerator для получения:

из одного входного изображения (на самом деле, это сцепленные 2 изображения, так что размер hxwx 6)

для нескольких выходных изображений (2 hxwx 3 изображения, 1 hxwx 1 изображение). Я хочу использовать 1,2M из 128 x 128 изображений, которые не могут быть загружены с моим предыдущим кодом.

мой код, не использующий ImageDataGenerator, был:

input,output1,output2,output3 = LoadImages()  #list of images 

img_in = keras.Model.Input((None,None,6))
out1,out2,out3 = MyNetwork(img_in)

mynet = keras.Model.Model(inputs = img_in, outputs = [out1,out2,out3])
mynet.compile(...compile parameters)
mynet.fit(input,[output1,output2,output3],batch_size=32,shuffle=True,validation_split=0.05)

Сейчас я изучаю, как использовать но не могу получить никакой подсказки для моей желаемой конфигурации. Пожалуйста, научите меня, как и большое спасибо.

...