如何遍历目录中的图像？

Question

我在目录中有猫图像。我想手动对这些图像应用数据增强。

我的目录是Downloads/Images/Cats/.
该目录包含带有两种标签/名称的图像：

1. cat1_001.png, cat1_002.png, ...

2. cat2_001.png, cat2_002.png, ...

指定两种类型的猫图像。

那么，如何仅从 python 目录中遍历标签为 1 的图像呢？

Answer 1

你可以这样做：import osdirectory_in_str = 'Downloads/Images/Cats/'directory = os.fsencode(directory_in_str)    for file in os.listdir(directory):    filename = os.fsdecode(file)    if(filename.startswith('cat1_') and filename.endswith('.png')):        print(filename)        # do your staff here