Конвертировать генератор в DirectoryIterator в Керасе - PullRequest
Новая
       7

Конвертировать генератор в DirectoryIterator в Керасе

0 голосов
/ 02 декабря 2018

Как я могу преобразовать генератор в DirectoryIterator в Keras ??

Я создал модель глубокого обучения с несколькими входами, и многие функции я не смог использовать, потому что генератор тестов являетсягенератор не DirectoryIterator. 

 input_imgen = ImageDataGenerator( 
                                  rotation_range=10,
                                  shear_range=0.2,
                                  zoom_range=0.1,
                                  width_shift_range=0.1,
                                  height_shift_range=0.1
                                  )



test_imgen = ImageDataGenerator()



def generate_generator_multiple(generator,dir1, dir2, batch_size, img_height,img_width):


    genX1 = generator.flow_from_directory(dir1,
                                          target_size = (img_height,img_width),
                                          class_mode = 'categorical',
                                          batch_size = batch_size,
                                          shuffle=False, 
                                          seed=7)

    genX2 = generator.flow_from_directory(dir2,
                                          target_size = (img_height,img_width),
                                          class_mode = 'categorical',
                                          batch_size = batch_size,
                                          shuffle=False, 
                                          seed=7)
    while True:
      X2i = genX2.next() 
      X1i = genX1.next() 

      yield [X1i [0], X2i [0]], X1i [1]  #Yield both images and their mutual label


inputgenerator=generate_generator_multiple(generator=input_imgen,
                                           dir1=dira,
                                           dir2=dirb,
                                           batch_size=32,
                                           img_height=224,
                                           img_width=224)       

testgenerator=generate_generator_multiple(generator=test_imgen,
                                          dir1=dirc,
                                          dir2=dird,
                                          batch_size=32,
                                          img_height=224,
                                          img_width=224)

Я не мог использовать следующие функции, потому что он получает ошибки. 

label2index = testgenerator.classes
AttributeError: 'generator' object has no attribute 'classes'

label2index = testgenerator.class_indices
AttributeError: 'generator' object has no attribute 'class_indices'

Я хочу преобразовать генератор в DirectoryIterator, чтобы использовать функции выше.

...