使用python將圖片按標簽分入不同文件夾的方法
給定圖像集如下,所有類別的圖片均在一個文件夾內(nèi):
給定與圖片名相匹配的表格,聲明每張圖片對應(yīng)的類別(共有20個類別):
那么,如何根據(jù)表格中所給的類別將圖片分入對應(yīng)的文件夾內(nèi)呢?以我的情況為例,我想將圖片分為20類(CATEGORY_ID有0-19共20類),可利用下面的代碼進行分類(經(jīng)細心網(wǎng)友指正,代碼已做出修改)。
#引入相關(guān)庫 import pandas as pd import os import shutil #用于移動文件
#打開表格文件并讀取 f=open("list.csv","rb") #輸入表格所在路徑+名稱 list=pd.read_csv(f) list["FILE_ID_JPG"]=".jpg" #建立圖片名與類別相對應(yīng) list["FILE_ID1"]=list["FILE_ID"]+list["FILE_ID_JPG"] #建立圖片名與類別相對應(yīng)
#創(chuàng)建文件夾 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將圖片按標簽分入不同文件夾的方法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
python編程實現(xiàn)清理微信重復(fù)緩存文件
這篇文章主要為大家介紹了使用python編程來實現(xiàn)清理微信重復(fù)緩存文件的示例代碼過程,有需要的朋友可以借鑒參考下,希望能夠有所幫助2021-11-11K-means聚類算法介紹與利用python實現(xiàn)的代碼示例
K-means聚類算法(事先數(shù)據(jù)并沒有類別之分!所有的數(shù)據(jù)都是一樣的)是我們大家應(yīng)該都聽過的一種算法,下面這篇文章主要給大家介紹了關(guān)于K-means聚類算法的基礎(chǔ)知識與利用python如何實現(xiàn)該算法的相關(guān)資料,需要的朋友可以參考借鑒,下面來一起看看吧。2017-11-11python3往mysql插入二進制圖片出現(xiàn)1064錯誤
這篇文章主要介紹了python3往mysql插入二進制圖片出現(xiàn)1064錯誤的解決方案,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2023-03-03python3新特性函數(shù)注釋Function Annotations用法分析
這篇文章主要介紹了python3新特性函數(shù)注釋Function Annotations用法,結(jié)合實例形式分析了Python3函數(shù)注釋的定義方法與使用技巧,需要的朋友可以參考下2016-07-07Python多進程編程multiprocessing代碼實例
這篇文章主要介紹了Python多進程編程multiprocessing代碼實例,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-03-03