python 從文件夾抽取圖片另存的方法

今天小編就爲大家分享一篇python 從文件夾抽取圖片另存的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧

有一個比較大的數據集需要自己處理,在分出訓練集和測試集時,如果靠手動實在太麻煩,於是自己寫了一段代碼。(其實就是在某一路徑下的子文件夾裏取出符合要求的圖片,放到另一個路徑的對應文件夾中)

from PIL import Image
import os
import glob
 
def read_image(cwd,newpath):
 for roots,dirs,files in os.walk(cwd):
  print(dirs)
  for i in dirs:
   print(i)
   os.chdir(cwd+i)
   for pic in glob.glob('*.jpg'):
    _,image=pic.split('_')
    img=image.split('.')[0]
    print(img)
    if len(img)!=0:
     if int(img)%2!=0:
      im=Image.open(pic)
      im.save(newpath+i+'/'+pic)

調用的時候直接

read_image('/media/fly/4898FC1598FC02EC/RAN/AID_dataset/AID_dataset/AID/','/media/fly/4898FC1598FC02EC/RAN/AID_dataset/AID_dataset/testing/')

這是在AID文件夾下的所有子文件夾中取出編號爲奇數的圖片,放到另一個路徑的同名文件夾下。我的圖片命名方法是Airport_1.jpg.

不一定要取出一半圖片,這個可以自行修改

以上這篇python 從文件夾抽取圖片另存的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持神馬文庫。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章