我正在尝试使用图像生成器训练模型,但它在 5 个类中给了我 0 个图像虽然子文件夹中有图像,但图像是 .dcm 格式
下面是我的代码
将我们的数据增强参数添加到 ImageDataGenerator
train_datagen = ImageDataGenerator(rescale = 1./255.,rotation_range = 40, width_shift_range = 0.2, height_shift_range = 0.2,shear_range = 0.2, zoom_range = 0.2, Horizontal_flip = True)
train_generator = train_datagen.flow_from_directory(TRAIN_PATH, batch_size = 20, class_mode = 'binary', target_size = (224, 224))
结果我找到了属于 5 个类的 0 个图像。