如何遍历目录中的图像？

Question

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

我的目录为php artisan optimize:clear。
此目录中的图像带有两种类型的标签/名称：

1. Downloads/Images/Cats/，cat1_001.png，...
2. cat1_002.png，cat2_001.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