使用 Keras flow_from_directory 和我的方法的预处理方法是否相同？答案

【问题标题】：Is it same preprocessing method using Keras flow_from_directory and my method?使用 Keras flow_from_directory 和我的方法的预处理方法是否相同？
【发布时间】：2020-08-07 07:14:26
【问题描述】：

这是使用 ImageDataGenerator 和 flow_from_directory 在 Keras 中训练我的模型的预处理。

train_datagen = ImageDataGenerator(rescale = 1./255)

train_dir = os.path.join(r'C:\Users\Admin\Desktop\deeplearning\ajou cat project resnet 18\data\trainset')
 
train_generator = train_datagen.flow_from_directory(train_dir, batch_size=16, target_size=(224, 224), color_mode='rgb')

# number of classes
K = 10

input_tensor = Input(shape=(224, 224, 3), dtype='float32', name='input')

#
# model architecture
#

x = conv1_layer(input_tensor)
x = conv2_layer(x)
x = conv3_layer(x)
x = conv4_layer(x)
x = conv5_layer(x)
x = GlobalAveragePooling2D()(x)
output_tensor = Dense(K, activation='softmax')(x)
 
resnet = Model(input_tensor, output_tensor)

resnet.compile(loss = 'categorical_crossentropy', optimizer = 'rmsprop', metrics = ['accuracy'])
    
resnet.fit(train_generator, steps_per_epoch = 11, epochs = 50)

这是我为用一张图片测试我的训练模型而制作的预处理方法。

target = (224, 224)
image = pilimg.open(r'C:\Users\Admin\Desktop\deeplearning\ajou cat project resnet 18\test\test.jpg')
image = image.resize(target)
image = img_to_array(image)
image = np.expand_dims(image, axis = 0)
image = image/255

它们是相同的预处理吗？

【问题讨论】：

标签： python tensorflow keras

【解决方案1】：

是的，它们是相同的，但如果您想应用多个处理并确定处理方法，ImageDataGenerator 采用preprocessing_function 参数。您可以定义一个预处理函数并将其提供给生成器。然后在测试时您可以使用相同的功能。

【讨论】：