給定圖像集如下,所有類別的圖片均在一個文件夾內:
給定與圖片名相匹配的表格,聲明每張圖片對應的類別(共有20個類別):
那么,如何根據表格中所給的類別將圖片分入對應的文件夾內呢?以我的情況為例,我想將圖片分為20類(CATEGORY_ID有0-19共20類),可利用下面的代碼進行分類(經細心網友指正,代碼已做出修改)。
#引入相關庫import pandas as pdimport osimport shutil #用于移動文件
#打開表格文件并讀取f=open("list.csv","rb") #輸入表格所在路徑+名稱list=pd.read_csv(f)list["FILE_ID_JPG"]=".jpg" #建立圖片名與類別相對應list["FILE_ID1"]=list["FILE_ID"]+list["FILE_ID_JPG"] #建立圖片名與類別相對應#創建文件夾for i in range(20): os.mkdir(str(i))
#進行分類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將圖片按標簽分入不同文件夾的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持VEVB武林網。
新聞熱點
疑難解答