【发布时间】:2021-08-13 13:35:52
【问题描述】:
我已经使用下面的代码从我的目录中加载图像
from keras.preprocessing.image import ImageDataGenerator
DATA_TRAIN_PATH='/content/drive/MyDrive/persiandataset/PersianDataset/train'
DATA_VALIDATION_PATH ='/content/drive/MyDrive/persiandataset/PersianDataset/validation'
train_datagen = ImageDataGenerator()
IMG_WIDTH =32
IMG_HEIGHT =32
EPOCHS_VGG=60
BATCH_SIZE_VGG=500
train_generator = train_datagen.flow_from_directory(
DATA_TRAIN_PATH,
target_size=(IMG_WIDTH, IMG_HEIGHT),
batch_size=BATCH_SIZE_VGG,
subset = "training",
class_mode='categorical')
test_generator = train_datagen.flow_from_directory(
DATA_VALIDATION_PATH,
target_size=(IMG_WIDTH, IMG_HEIGHT),
batch_size=BATCH_SIZE_VGG,
subset = "validation",
class_mode='categorical')
结果显示: 找到属于 32 个类的 37109 张图片。
找到 0 张图片属于 32 个类别。
但我已经为我的验证数据正确定义了子目录,定义训练目录和验证目录之间没有区别,但它无法识别我在验证文件夹中的图像。 我该如何解决这个问题? 我还附上了我的文件夹图片subdirectory for images
我的验证目录包含 32 个类,每个子目录中有 27 张图像。所有验证数据都等于 864。
【问题讨论】:
-
图片的扩展名是什么?
-
@Kaveh 这是“.tif”
标签: python keras google-colaboratory tf.keras