如何在 Pytorch 的同一文件夹中加载图像？答案

【问题标题】：How to load images in the same folder in Pytorch?如何在 Pytorch 的同一文件夹中加载图像？
【发布时间】：2019-02-27 13:58:44
【问题描述】：

我想分别从文件夹/img 和/mask 加载所有图像。数据结构如下：

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

希望能得到帮助。

【问题讨论】：

你想在哪里加载图片，你想把图片从一个文件夹加载到另一个文件夹吗？

标签： python computer-vision pytorch

【解决方案1】：

如果你想加载这两个文件夹中的所有图片，那么你可以试试 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))

【讨论】：

谢谢，顺便问一下，我可以使用DataLoader 或其他类来加载它吗？如何加载？我想在多个线程中输入图像。
查看链接是否有帮助 how_do_i_load_images_into_pytorch_for_training 或 how-do-you-load-images-into-pytorch-dataloader

【解决方案2】：

如果你的 Python 是 3.4+，您可以使用pathlib获取具有相同扩展名的文件列表。

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

结果：

[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'),
 ...
]

【讨论】：