下面是在深度學習數據集處理過程中可能會用到的一個小程序,幫助我們根據圖片文件的名字來分開圖片:
import osimport shutilpath_img='讀取圖片的路徑'ls = os.listdir(path_img)print(len(ls))for i in ls: if i.find('查找的關鍵詞')!=-1: shutil.move(path_img+'/'+i,"輸出保存的路徑"+i)
—————————使用示例——————————
例如我有如下的一個圖片集,可以看到男性的圖片都包含有“testnan”,女性的圖片都包含“testnv”,現在想要把兩個數據集分開獨立的打標簽,就可以用上面的小程序了。
import osimport shutilpath_img='C://Users//chriszhang//Desktop//gender//test'ls = os.listdir(path_img)print(len(ls))for i in ls: if i.find('testnan')!=-1: shutil.move(path_img+'/'+i,"C:/Users/chriszhang/Desktop/male/"+i)
如上就可以將我桌面上gender/test文件夾中的男性圖片轉到桌面上的male文件夾中了。
以上這篇Python根據文件名批量轉移圖片的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答