keras调用flow_from_directory()出现“Found 0 images belonging to 2 classes”问题

在使用Keras进行图像分类时,遇到'Found 0 images belonging to 2 classes'错误。问题源于train_datagen中的validation_split参数与后续验证集设置的冲突。解决方案包括:1) 从训练集中直接分出验证集;2) 单独创建验证集目录时,不使用validation_split参数。此外,介绍了model.fit_generator()和model.fit()在fit方法中的区别,包括validation_split、validation_data参数的作用。
摘要由CSDN通过智能技术生成

图像分类时,keras调用flow_from_directory()出现“Found 0 images belonging to 5 classes”问题

代码如下:

from tensorflow import keras
from keras_preprocessing import image


train_datagen = image.ImageDataGenerator(
    #..... 
    fill_mode = 'nearest',   
    validation_split=0.3
   
) 
valid_datagen = image.ImageDataGenerator(
    rescale=1./255,
)
test_datagen = image.ImageDataGenerator(
    rescale = 1./255,
)


train_generator = train_datagen.flow_from_directory(
    'zuizhong/train',
    target_size=(224,224),#(height,width)
    batch_size=batch_size,
    #subset='training',
    )
valid_generator = valid_datagen.flow_from_directory(
    'zuizhong/val',
    target_size=(224,224),
    batch_size=batch_size,
    subset='validation',
    

    )
#0: 'jinyu118', 1: 'kenuo58', 2: 'liyuan296', 3: 
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值