2019-03-07

2019-03-07  本文已影响0人  温柔乡_英雄冢

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

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

首先是os.walk()函数:

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

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

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

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

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

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

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

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

上一篇下一篇

猜你喜欢

热点阅读