在圖片處理過(guò)程中弊知,樣本數(shù)據(jù)的組織是個(gè)常見(jiàn)的問(wèn)題懦鼠,樣本組織好了纪铺,后面數(shù)據(jù)轉(zhuǎn)換相速、讀取等都能得心應(yīng)手。
在這里講一下我在準(zhǔn)備樣本時(shí)的一些心得;
首先是os.walk()函數(shù):
os.walk(path)返回三個(gè)值:parent,dirnames,filenames霹陡,分別表示path的路徑和蚪、path路徑下的文件夾的名字和path路徑下文件夾以外的其他文件;
我用下面的代碼來(lái)對(duì)樣本圖片建立標(biāo)簽:
不過(guò)不知walk到底是怎么遍歷的止状,我生成的標(biāo)簽順序亂了
所以這就想到用listdir來(lái)重新做標(biāo)簽:
os.listdir(path)返回一個(gè)包含path路徑下所有文件的列表
我用下面的方法來(lái)建立標(biāo)簽:
不過(guò)還是之前一樣的亂序烹棉,最后采用sort():
先獲取文件夾下圖片文件的名字列表list,再對(duì)圖片的序號(hào)進(jìn)行排序怯疤,將排序好的文件名列表逐一寫(xiě)入txt中