python - 如何为图像分割模型创建数据生成器？

Question

我正在研究图像分割问题。我的数据集中有 3 种类型的图像（Drishti_GS）。一个是原始眼底图像，另外两个是软图图像，即视杯段和视盘段。我正在尝试制作一个数据生成器，我可以在上面训练我的模型。我附上了使用以下代码后获得的图像名称的屏幕截图。

import os

for root, dirs, files in os.walk("."):
    for filename in files:
      if filename.endswith(".png"):

        print(filename)

我需要加载这些图像。我希望有人可以帮助我提供一些具体的代码或一些有用的材料。

标签： pythonfileimage-processing

Answer 1

您可以通过以下方式加载图像：

for image_name in os.listdir("data_path"):
    image = keras.preprocessing.image.load_img('data_path/'+image_name, color_mode='rgb', target_size=(128,128))
    names = image_name.split('_')
    if names[-1]=="cupsegSoftmap.png":
        label = ...
    elif names[-1]=="ODsegSoftmap.png":
        label = ...

您可以手动将标签值设置为数组