【发布时间】:2020-08-02 10:09:22
【问题描述】:
model=tf.keras.Sequential(
[
Conv2D(filters=1, kernel_size=(3, 3), activation='relu', strides=1, padding='same'),
MaxPooling2D(pool_size=(4, 4)),
Flatten(),
Dense(6,activation="softmax")
])
model.compile(loss='CategoricalCrossentropy',
optimizer='adam',
metrics=['accuracy'])
train_datagen = ImageDataGenerator( # Real time data augmentation
rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
train_generator = train_datagen.flow_from_directory( # Loads the File which contains the images.
'train', # Returns x (numpy array containing a batch of images)
target_size=(150, 150), # and correcponding labels
class_mode='categorical')
model.fit( train_generator,
batch_size=100,
epochs=1)
此代码的输出不提供有关数据增强的任何信息。那么谁能澄清一下这段代码是否增加了可用数据?
【问题讨论】:
-
train_datagen = ImageDataGenerator( # 实时数据增强 rescale=1./255,shear_range=0.2, zoom_range=0.2, Horizontal_flip=True) 这增强了输入数据
标签: python tensorflow keras conv-neural-network data-augmentation