import os from PIL import Image import glob def convertjpg(jpgfile,outdir,width=200,height=500): img=Image.open(jpgfile) new_img=img.resize((width,height),Image.BILINEAR) new_img.save(os.path.join(outdir,os.path.basename(jpgfile))) for jpgfile in glob.glob(('/home/yangguide/Videos/images/*.png')): convertjpg(jpgfile,"/home/yangguide/Videos/image_2")
圖片庫PIL(Python Image Library)是python的第三方圖像處理庫,但是由於其強大的功能與眾多的使用人數,幾乎已經被認為是python官方圖像處理庫了。
Image類別是PIL中的核心類,你有很多種方式來對它進行初始化,例如從檔案載入一張圖像,處理其他形式的圖像,或是從頭創造一張圖像等。
Image模組操作的基本方法都包含於此模組內。如open、save、conver、show…等方法。
1.載入映像,使用Image類別的open()函數:
Image.open(jpgfile)
2.儲存映像,使用Image類別的save()函數:
new_img.save(os.path.join(outdir,os.path.basename(jpgfile)))
3.os.path.basename()方法:
傳回path最後的檔名, 如果path以’/'結尾,那麼就會回傳空值, 即os.path.split(path)的第二個元素。
範例:
>>> import os >>> path = '/Users/beazley/Data/data.csv' >>> os.path.basename(path) #Get the last component of the path 'data.csv'
4.img.resize((width,height),Image.BILINEAR) :
使用resize函數指定映像的大小和質量,第二個參數設定和意義如下圖:
#5.glob.glob()與glob.iglob()的用法:
glob.glob()可同時取得所有的符合路徑,而glob.iglob()一次只能取得一個符合路徑。
分享一個比較實用的功能,改一下檔案路徑和縮放尺寸即可適配成自己的。
適用於原來是不同尺寸的圖片,不好統一縮放的,只能放到一張統一大小的畫布里。
如果原來的圖片尺寸是一致的,請參考本人另一篇博客,自己找一下咯。
運行環境:python3.5
需要安裝一下opencv,如果有anaconda,執行conda install opencv-python
# *_* coding : UTF-8 *_* # 开发人员 :csu·pan-_-|| # 开发时间 :2020/11/09 16:40 # 文件名称 :renameFile.py # 开发工具 :PyCharm # 功能描述 :将文件夹下的图片全部缩放,裁减,并按新文件名存储 import os import cv2 path = 'E:/Projects/images' # 原文件夹路径 newpath = 'E:/Projects/newimages' # 新文件夹路径 files = os.listdir(path) # 获取文件名列表 for i, file in enumerate(files): # 展开文件名的列表和索引 if file.endswith('.jpg'): imgName = os.path.join(path, file) # 获取文件完整路径 img = cv2.imread(imgName) # 读图 imgNew = cv2.resize(img, (1200, 1200)) # 缩放 # imgNew = imgNew[60:552,:] # 截取一部分区域 newName = os.path.join(newpath, 'img_%03d'%(0+i)+'.jpg') # 设置新的文件名 print(newName) cv2.imwrite(newName,imgNew) # 存储按新文件名命令的图片
後面來了新的需求,作為一個程式設計師,需求是永遠要去滿足的。縮放的同時,需要保持原有比例,全部設定到一張800 * 800的全黑畫布上面,即補零操作,又重新調整了程式碼,其中的核心部分就是判斷影像的長邊是否大於800,大於800則將800與長邊的比值設為縮放比例,小於800則原圖大小不變。需要匯入一個新函式庫numpy,conda install numpy
:
# *_* coding : UTF-8 *_* # 开发人员 :csu·pan-_-|| # 开发时间 :2020/11/09 18:15 # 文件名称 :renameFile.py # 开发工具 :PyCharm # 功能描述 :将文件夹下的图片全部缩放(同时保持原有宽高比例),裁切,并按新文件名存储 import os import cv2 import numpy as np path = r'E:\Projects\images' # 原文件夹路径 newpath = r'E:\Projects\newimages' # 新文件夹路径 files = os.listdir(path) # 获取文件名列表 c_w ,c_h = 800,800 # 全黑画布的大小 for i, file in enumerate(files): img_zeros = np.zeros((c_w, c_h, 3), np.uint8) # 创建全黑的图像 if file.endswith('.jpg'): imgName = os.path.join(path, file) # 获取文件完整路径 img = cv2.imread(imgName) # 读图 h, w , _ = img.shape # 获取图像宽高 # 缩放图像,宽高大于800的按长边等比例缩放,小于800的保持原图像大小: if max(w,h) > c_w: ratio = c_w / max(w,h) imgcrop = cv2.resize(img, (round(w * ratio) , round(h * ratio))) # 将裁切后的图像复制进全黑图像里 img_zeros[0:round(h * ratio), 0:round(w * ratio)] = imgcrop else: img_zeros[0:h, 0:w] = img # imgNew = imgNew[60:552,:] # 截取一部分 # 设置新的文件名: newName = os.path.join(newpath, 'img_%03d'%(0+i)+'.jpg') print(newName) cv2.imwrite(newName,img_zeros) # 存储按新文件名命令的图片
標註的xml檔需要同步修改,於是把程式碼調整了一下:
# *_* coding : UTF-8 *_* # 开发人员 :csu·pan-_-|| # 开发时间 :2020/11/09 18:15 # 文件名称 :renameFile.py # 开发工具 :PyCharm # 功能描述 :将文件夹下的图片全部缩放(同时保持原有宽高比例),裁切,并按新文件名存储 # 同时调整xml里的坐标信息 import os import cv2 import numpy as np import xml.etree.ElementTree as ET path = r'C:\Users\Administrator\Desktop\test' # 原文件夹路径 newpath = r'C:\Users\Administrator\Desktop\newtest' # 新文件夹路径 c_w ,c_h = 800,800 # 全黑画布的大小 def edit_xml(xml_file,ratio,i): """ 修改xml文件 :param xml_file:xml文件的路径 :return: """ all_xml_file = os.path.join(path, xml_file) tree = ET.parse(all_xml_file) objs = tree.findall('object') for ix, obj in enumerate(objs): type = obj.find('type').text if type == 'bndbox': obj_bnd = obj.find('bndbox') obj_xmin = obj_bnd.find('xmin') obj_ymin = obj_bnd.find('ymin') obj_xmax = obj_bnd.find('xmax') obj_ymax = obj_bnd.find('ymax') xmin = float(obj_xmin.text) ymin = float(obj_ymin.text) xmax = float(obj_xmax.text) ymax = float(obj_ymax.text) obj_xmin.text = str(round(xmin * ratio)) obj_ymin.text = str(round(ymin * ratio)) obj_xmax.text = str(round(xmax * ratio)) obj_ymax.text = str(round(ymax * ratio)) elif type == 'robndbox': obj_bnd = obj.find('robndbox') obj_cx = obj_bnd.find('cx') obj_cy = obj_bnd.find('cy') obj_w = obj_bnd.find('w') obj_h = obj_bnd.find('h') obj_angle = obj_bnd.find('angle') cx = float(obj_cx.text) cy = float(obj_cy.text) w = float(obj_w.text) h = float(obj_h.text) obj_cx.text = str(cx * ratio) obj_cy.text = str(cy * ratio) obj_w.text = str(w * ratio) obj_h.text = str(h * ratio) newfile = os.path.join(newpath, '%05d'%(0+i)+'.xml') tree.write(newfile, method='xml', encoding='utf-8') # 更新xml文件 if __name__ == '__main__': files = os.listdir(path) # 获取文件名列表 for i, file in enumerate(files): img_zeros = np.zeros((c_w, c_h, 3), np.uint8) # 创建全黑的图像 if file.endswith('.jpg'): imgName = os.path.join(path, file) # 获取文件完整路径 xml_file = file.replace('.jpg','.xml') img = cv2.imread(imgName) # 读图 h, w , _ = img.shape # 获取图像宽高 # 缩放图像,宽高大于800的按长边等比例缩放,小于800的保持原图像大小: if max(w,h) > c_w: ratio = c_w / max(w,h) imgcrop = cv2.resize(img, (round(w * ratio) , round(h * ratio))) # 将裁切后的图像复制进全黑图像里 img_zeros[0:round(h * ratio), 0:round(w * ratio)] = imgcrop edit_xml(xml_file, ratio, i) else: img_zeros[0:h, 0:w] = img edit_xml(xml_file, 1, i) # 设置新的文件名: newName = os.path.join(newpath, '%05d'%(0+i)+'.jpg') print(newName) cv2.imwrite(newName,img_zeros) # 存储按新文件名命令的图片
以上是如何使用Python批次修改影像大小,統一大小?的詳細內容。更多資訊請關注PHP中文網其他相關文章!