图像分类时,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: