2019-03-07

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

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

首先是os.walk()函数:


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

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

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


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

os.listdir(path)返回一个包含path路径下所有文件的列表


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


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


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

你可能感兴趣的:(2019-03-07)