0

我正在尝试使用图像生成器训练模型,但它在 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, Horizo​​ntal_flip = True)

train_generator = train_datagen.flow_from_directory(TRAIN_PATH, batch_size = 20, class_mode = 'binary', target_size = (224, 224))

结果我找到了属于 5 个类的 0 个图像。

4

0 回答 0