繁体   English   中英

Keras ImageDataGenerator不同的图像

[英]Keras ImageDataGenerator different images

我在使用Keras的ImageDataGenerator进行图像增强时遇到了麻烦。 现在,我正在尝试垂直翻转训练数据集中的图像。 X_batch是我的翻转图像数据集,而X_train是我的原始训练数据集。

有人可以解释为什么在图像X_batch都以不同的顺序比图像X_train X_batch[0]应该是一个翻转版本X_train[0]而是X_batch[0]是一个不同的形象在我的数据集翻转版本。

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)    

datagen = ImageDataGenerator(vertical_flip=True)
datagen.fit(X_train)

for X_batch, y_batch in datagen.flow(X_train, y_train):
    X_batch = X_batch.astype('uint8')

    plt.subplot(2, 1, 1)
    plt.imshow(X_batch[0]) // flipped image

    plt.subplot(2, 1, 2)
    plt.imshow(X_train[0]) // original image

    plt.show()
    break

根据Keras文档flow方法采用名为shuffle的参数,如果将其设置为True (默认情况下),则将对数据进行shuffle ,然后应用图像转换。 如果您不喜欢这种行为,可以将其设置为False

for X_batch, y_batch in datagen.flow(X_train, y_train, shuffle=False):
    ...

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM