【问题标题】:flow_from_directory seems to find no images but correctly recognized classesflow_from_directory 似乎找不到图像但正确识别的类
【发布时间】: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


【解决方案1】:

设置subset="training"subset="validation" 可能会导致问题,因为您没有在ImageDataGenerator() 类中设置validation_split。您可以尝试不设置这样的参数吗:

test_generator = train_datagen.flow_from_directory(
        DATA_VALIDATION_PATH,
        target_size=(IMG_WIDTH, IMG_HEIGHT),
        batch_size=BATCH_SIZE_VGG,
        class_mode='categorical')

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2014-11-30
    • 2015-01-07
    • 2021-04-12
    • 1970-01-01
    • 2014-05-20
    • 1970-01-01
    • 2011-01-12
    相关资源
    最近更新 更多