在图片处理过程中,样本数据的组织是个常见的问题,样本组织好了,后面数据转换、读取等都能得心应手。
在这里讲一下我在准备样本时的一些心得;
首先是os.walk()函数:
os.walk(path)返回三个值:parent,dirnames,filenames,分别表示path的路径、path路径下的文件夹的名字和path路径下文件夹以外的其他文件;
我用下面的代码来对样本图片建立标签:
不过不知walk到底是怎么遍历的,我生成的标签顺序乱了
所以这就想到用listdir来重新做标签:
os.listdir(path)返回一个包含path路径下所有文件的列表
我用下面的方法来建立标签:
不过还是之前一样的乱序,最后采用sort():
先获取文件夹下图片文件的名字列表list,再对图片的序号进行排序,将排序好的文件名列表逐一写入txt中
网友评论