Python文件遍历os.walk()与os.listdir()

在图片处理过程中,样本数据的组织是个常见的问题,样本组织好了,后面数据转换、读取等都能得心应手。

在这里讲一下我在准备样本时的一些心得;

首先是os.walk()函数:

os.walk(path)返回三个值:parent,dirnames,filenames,分别表示path的路径、path路径下的文件夹的名字和path路径下文件夹以外的其他文件;

我用下面的代码来对样本图片建立标签:

Python文件遍历os.walk()与os.listdir()_第1张图片

不过不知walk到底是怎么遍历的,我生成的标签顺序乱了

Python文件遍历os.walk()与os.listdir()_第2张图片

所以这就想到用listdir来重新做标签:

os.listdir(path)返回一个包含path路径下所有文件的列表Python文件遍历os.walk()与os.listdir()_第3张图片

我用下面的方法来建立标签:

Python文件遍历os.walk()与os.listdir()_第4张图片

不过还是之前一样的乱序,最后采用sort():

Python文件遍历os.walk()与os.listdir()_第5张图片

先获取文件夹下图片文件的名字列表list,再对图片的序号进行排序,将排序好的文件名列表逐一写入txt中




你可能感兴趣的:(图像处理)