給定圖像集如下，所有類別的圖片均在一個文件夾內(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)容了，希望能給大家一個參考，也希望大家多多支持。

總結(jié)

以上是生活随笔為你收集整理的python根据文件名打标签_使用python将图片按标签分入不同文件夹的方法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。