代码如下:
from keras.preprocessing.image import ImageDataGenerator
train_datagen=ImageDataGenerator(rescale=1./255)
test_datagen=ImageDataGenerator(rescale=1./255)
train_generator=test_datagen.flow_from_directory(
'D:/BaiduNetdiskDownload/kaggle/cats_and_dogs_small/train/cats',
target_size=(150,150),
batch_size=20,
class_mode='binary'
)
validation_generator=test_datagen.flow_from_directory(
'D:/BaiduNetdiskDownload/kaggle/cats_and_dogs_small/validation/cats',
target_size=(150,150),
batch_size=20,
class_mode='binary'
)
for data_batch,labels_batch in train_generator:
print('data batch shape:',data_batch.shape)
print('labels batch shape:',labels_batch.shape)
break
最后结果如下:
Found 0 images belonging to 0 classes.
Found 0 images belonging to 0 classes.
data batch shape: (0, 150, 150, 3)
labels batch shape: (0,)
Process finished with exit code 0
应该怎么解决啊
相似问题