深度学习中Python常用操作

文件夹下所有特定格式文件

查找 ‘./data/images’ 文件夹下所有 png格式图片。

import os

data_dir = './data/images'
filename = [name for name in os.listdir(data_dir) if os.path.splitext(name)[-1] == '.png']