也许我理解错了。如果我在我的ImageDataGenerator中实现channel_shift_range,输出应该具有“混乱”的颜色值,对吧?我想使用它来使我的模型更能应对颜色变化的差异。
然而,当我测试它时,我没有看到任何效果。我是不是用错了?这是我的代码:
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-Preprocessing==1.0.9
tensorboard==1.9.0
tensorflow-gpu==1.9.0
提前感谢您的帮助!