读取多个文件夹中的图像

Question

我在读取多个文件的图像时遇到问题。我有一个这样的文件：

现在，我想读取 rgb 文件中的所有图像，但 rgb 文件中还有 3 个文件夹 (00,01,02)，其中包含图像。如何读取 3 个文件夹 00、01、02 中的所有图像。此外，在文件夹注释中，我也有 3 个文件夹，分别命名为 00、01 和 02，但是在这 3 个文件夹中，它包含 2 种文件，即 txt 和 npy。我只想看txt文件，怎么办？
我将使用“rgb”文件中的图像来训练模型，txt 文件将用作标签

Answer 1

pathlib.Path() 允许通配符匹配您要查找的内容。 ** 是匹配“所有子目录”的通配符，* 匹配文件名中的任何内容：

from pathlib import Path

p = Path('Videos frame')

for f in p.glob('rgb/**/*.png'): # or whatever your image type is...
    print(f)

for f in p.glob('annotation/**/*.txt'):
    print(f)

读取多个文件夹中的图像

Reading images in multiple folders

python

file

path