Как загрузить изображения и маску для unet, с помощью keras imagedatagenerator - PullRequest
1 голос
/ 25 сентября 2019

Мне нужно загрузить изображения и маску для модели unet, я не хочу использовать cv2 или любые библиотеки для загрузки изображений. Я хочу использовать простой способ загрузки этих изображений.Мой набор данных Strucuture

data/train
    1.png
    2.png

data/mask
    1_mask.png

во время поиска в Google я нашел документы keras для изображений и маски кода. Но ему нужны изображения и маска, хранящиеся в массиве. Или любой другой метод загрузки изображений и маски.

и как работать, если ниже дело.

/train:
      /dir_name/
                /images
                /masks
      /dir_name2/
                /images
                /masks
      /dir_name3/
                /images
                /masks
      /dir_name4/
                /images
                /masks
...