我正在使用 ImageDataGenerator 来增强我的图像。我需要从生成器中获取 y 标签。
示例:我有 10 个训练图像,7 个是标签 0,3 个是标签 1。我想将训练集大小增加到 100。
total_training_images = 100 total_val_images = 50
model.fit_generator(
train_generator,
steps_per_epoch= total_training_images // batch_size,
epochs=epochs,
validation_data=validation_generator,
validation_steps= total_val_images // batch_size)
据我了解,这会在每个 epoch 的 100 个训练图像上训练一个模型,每个图像都会根据我的数据生成器以某种方式进行增强,然后在 50 个图像上进行验证。
如果我这样做train_generator.classes
,我会得到一个输出 [0,0,0,0,0,0,0,1,1,1]。这对应于我的标签 0 的 7 个图像和标签 1 的 3 个图像。
对于这 100 张新图像,我如何获得 y 标签?这是否意味着当我将其增加到 100 张图像时,我的新train_generator
标签是相同的,但重复了 10 次?基本上np.append(train_generator.classes)
10倍?
如果有帮助,我正在关注本教程: https ://blog.keras.io/building-powerful-image-classification-models-using-very-little-data.html