时间戳

首页 > 解决方案 > random_transform() 缺少 1 个必需的位置参数：'x'

python - random_transform() 缺少 1 个必需的位置参数：'x'

问题描述

我尝试使用random_transform来自 tensorflow 的方法为图像生成 6 个随机变换。这是我的代码：

train_image_generator = ImageDataGenerator(rescale=1./255)
train_data_gen = train_image_generator.flow_from_directory(
                               train_dir, 
                               target_size=(IMG_HEIGHT, IMG_WIDTH),
                               color_mode='rgb',
                               classes=None, 
                               class_mode='categorical', 
                               batch_size=128, 
                               shuffle='True', 
                               seed=None, 
                               save_to_dir=None, 
                               save_prefix='', 
                               save_format='png', 
                               follow_links=False, 
                               subset=None, 
                               interpolation="nearest")
train_image_generator = train_image_generator.random_transform(
                                              train_data_gen[0][0][0], 
                                              seed=None)

我收到以下错误：

random_transform() missing 1 required positional argument: 'x'

我train_data_gen[0][0][0]是一个 3D 数组 - 我不知道我的错误在哪里。

标签： pythontensorflowkeras

解决方案

random_transform是ImageDataGenerator对象的一种方法。所以，你需要像这样的东西：

train_image_generator = ImageDataGenerator()
train_image_generator.random_transform(train_data_gen[0][0][0], seed=None)

推荐阅读