Генератор Keras возвращает расширенные данные с разным диапазоном пикселей - PullRequest
Новая
       75

Генератор Keras возвращает расширенные данные с разным диапазоном пикселей

0 голосов
/ 10 апреля 2020

Я пытаюсь использовать увеличение данных на основе Keras, проблема в том, что я получаю диапазон пикселей для дополненных данных, не совпадает с исходным изображением. Мне нужно сохранить диапазон пикселей моих изображений, любой доступный параметр в ImageDataGenerator?

код: 

 datagen = ImageDataGenerator( 
        rotation_range = 40, 
        shear_range = 0.2, 
        zoom_range = 0.2, 
        horizontal_flip = True, 
       ) 

for m in range(len(X_train)):
    img = X_train[m]
    img = img.reshape((1, ) + img.shape)  
    img2 =Y_train[m]
    img2 = img2.reshape((1, ) + img2.shape)  

    i = 0
    for batch in datagen.flow(img2, batch_size = 1, 
                          save_to_dir ='/Users/macbookpro/Desktop/resultat/dd/',  
                          save_prefix ='image', save_format ='png',seed = np.random.seed(0)): 

            i += 1
            if i > 5: 
                break
...