How to load images in the same folder in Pytorch?

Question

I want to load all of the images from the folder /img and /mask respectively. The data structure can be shown as follows:

data
    img
        0.png
        1.png
        2.png
        3.png
        ...
   mask
       label_0.png
       label_1.png
       label_2.png
       ...

Hopefully for help.

where do you want to load the images, do you want to load images from one folder to another? — LOKE2707
– LOKE2707, Commented Feb 27, 2019 at 5:51

LOKE2707 · Accepted Answer · 2019-02-27 05:50:52Z

1

If you want to load all the images from the two folders then you can try cv2

import cv2

img = []
for i in range(n): # n = number of images in img folder
    img_path = f'~data\img\{i}.png' # replace ~ with full path 
    img.append(cv2.imread(img_path))

for i in range(n): # n = number of images in mask folder
    img_path = f'~data\mask\lable_{i}.png' # replace ~ with full path
    img.append(cv2.imread(img_path))

answered Feb 27, 2019 at 5:50

LOKE2707

3121 gold badge6 silver badges20 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

rosefun Over a year ago

Thanks, by the way, can I use DataLoader or other class to load it and how? I want to input the images in multiple threads.

LOKE2707 Over a year ago

check out the links if it helps how_do_i_load_images_into_pytorch_for_training or how-do-you-load-images-into-pytorch-dataloader

StarFox · Accepted Answer · 2019-02-27 05:55:00Z

1

If your Python is 3.4+, You can get the file list which has the same extension name using pathlib.

from pathlib import Path
# *.png
result = list( Path('./data/').rglob('*.[pP][nN][gG]') )
print(result)

Outcome:

[PosixPath('data/img/0.png'),
 PosixPath('data/img/0.png'),
 PosixPath('data/img/0.png'),
 ...
 PosixPath('data/mask/label_0.png'),
 PosixPath('data/mask/label_1.png'),
 PosixPath('data/mask/label_2.png'),
 ...
]

answered Feb 27, 2019 at 5:55

StarFox

324 bronze badges

Collectives™ on Stack Overflow

How to load images in the same folder in Pytorch?

2 Answers 2

2 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related