如何遍历目录中的图像？

Question

我在目录中有猫图像。我想手动对这些图像应用数据增强。我的目录是Downloads/Images/Cats/.该目录包含带有两种标签/名称的图像：cat1_001.png, cat1_002.png, ...cat2_001.png, cat2_002.png, ...指定两种类型的猫图像。那么，如何仅从 python 目录中遍历标签为 1 的图像呢？

Answer 1

你可以这样做：

import os

directory_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