Эффекты channel_shift_range в ImageDataGenerator (увеличение изображения Keras) - PullRequest
Новая
       23

Эффекты channel_shift_range в ImageDataGenerator (увеличение изображения Keras)

0 голосов
/ 01 июля 2019

Может быть, я неправильно понимаю. Если я реализую channel_shift_range в моем ImageDataGenerator, выходные данные должны иметь «зашифрованные» значения цвета, верно? Я хотел бы использовать его, чтобы сделать мою модель более устойчивой к изменению цвета.

Однако, когда я проверяю это, я не вижу никакого эффекта. Я использую это неправильно? Вот мой код: 

from keras.preprocessing.image import ImageDataGenerator

import cv2
import matplotlib.pyplot as plt
%matplotlib inline

path = '/mnt/Project/Imaging/samples'

datagen = ImageDataGenerator(channel_shift_range=0.9)

genObject = datagen.flow_from_directory(path,
                                        batch_size=1)

augs = []
i = 0

for batch in genObject:
    augs.append(batch)
    i += 1
    if i > 10:
        break

for item in augs:
    plt.imshow(item[0][0].astype('uint8'))
    plt.show()

Окружающая среда:
Jupyter Lab
Python 3.6.6
Keras == 2.2.4
Keras-Applications == 1.0.7
Keras-Препроцессирование == 1.0.9
tensorboard == 1.9.0
tensorflow-GPU == 1.9.0

Заранее спасибо за помощь!

...