給定圖像集如下，所有類別的圖片均在一個(gè)文件夾內(nèi)：

給定與圖片名相匹配的表格，聲明每張圖片對(duì)應(yīng)的類別（共有20個(gè)類別）：

那么，如何根據(jù)表格中所給的類別將圖片分入對(duì)應(yīng)的文件夾內(nèi)呢？以我的情況為例，我想將圖片分為20類（CATEGORY_ID有0-19共20類），可利用下面的代碼進(jìn)行分類（經(jīng)細(xì)心網(wǎng)友指正，代碼已做出修改）。

#引入相關(guān)庫(kù)import pandas as pdimport osimport shutil #用于移動(dòng)文件

#打開(kāi)表格文件并讀取f=open("list.csv","rb") #輸入表格所在路徑+名稱list=pd.read_csv(f)list["FILE_ID_JPG"]=".jpg" #建立圖片名與類別相對(duì)應(yīng)list["FILE_ID1"]=list["FILE_ID"]+list["FILE_ID_JPG"] #建立圖片名與類別相對(duì)應(yīng)

#創(chuàng)建文件夾for i in range(20): os.mkdir(str(i))

#進(jìn)行分類for i in range(0,20): listnew=list[list["CATEGORY_ID"]==i] l=listnew["FILE_ID1"].tolist() j=str(i) for each in l:  shutil.move(each,j)

大功告成！

以上這篇使用python將圖片按標(biāo)簽分入不同文件夾的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持武林站長(zhǎng)站。