【发布时间】:2019-07-01 18:15:08
【问题描述】:
也许我理解错了。如果我在 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 实验室
Python 3.6.6
Keras==2.2.4
Keras-应用程序==1.0.7
Keras-预处理==1.0.9
张量板==1.9.0
tensorflow-gpu==1.9.0
提前感谢您的帮助!
【问题讨论】:
标签: python-3.x tensorflow keras computer-vision tf.keras