python根据文件名打标签_使用python将图片按标签分入不同文件夹的方法
給定圖像集如下,所有類別的圖片均在一個文件夾內:
給定與圖片名相匹配的表格,聲明每張圖片對應的類別(共有20個類別):
那么,如何根據表格中所給的類別將圖片分入對應的文件夾內呢?以我的情況為例,我想將圖片分為20類(CATEGORY_ID有0-19共20類),可利用下面的代碼進行分類(經細心網友指正,代碼已做出修改)。
#引入相關庫
import pandas as pd
import os
import 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將圖片按標簽分入不同文件夾的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持。
總結
以上是生活随笔為你收集整理的python根据文件名打标签_使用python将图片按标签分入不同文件夹的方法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: STM32H743+CubeMX-定时器
- 下一篇: Python | threading02