相关疑难解决方法(0)

用于语义分割的 ImageDataGenerator

我正在尝试使用 Keras 进行语义分割,并且在尝试加载图像时使用flow_from_directory方法出现此错误。

Found 0 images belonging to 0 classes.
Found 0 images belonging to 0 classes.
Run Code Online (Sandbox Code Playgroud)

这是我的代码。

from tensorflow.keras.applications.resnet50 import preprocess_input
from tensorflow.keras.preprocessing.image import ImageDataGenerator

data_generator = ImageDataGenerator()
train_generator = data_generator.flow_from_directory(
                                        directory="../input/Training_dataset/Images",
                                        target_size=(IMG_SIZE, IMG_SIZE),
                                        batch_size=16,
                                        class_mode=None,
                                        classes=None
                                        )

mask_generator = data_generator.flow_from_directory(
    directory="../input/Training_dataset/Masks/all",
    class_mode=None,
    classes=None,
    batch_size = 1,
    )
Run Code Online (Sandbox Code Playgroud)

我已经阅读了这个问题,但解决方案在 Keras 中不起作用,用于语义分割,flow_from_directory() 错误

python keras tensorflow semantic-segmentation

6
推荐指数
1
解决办法
6161
查看次数

标签 统计

keras ×1

python ×1

semantic-segmentation ×1

tensorflow ×1