下面代码是批量读取图片,统一图片尺寸,并将其转换为统一格式RGB输出,保存。
from PIL import Image
import os.path
import glob
def convertjpg(jpgfile,outdir,width=500,height=500):
img=Image.open(jpgfile)
try:
new_img = img.resize((width, height), Image.BILINEAR)
if new_img.mode == 'P':
new_img = new_img.convert("RGB")
if new_img.mode == 'RGBA':
new_img = new_img.convert("RGB")
new_img.save(os.path.join(outdir, os.path.basename(jpgfile)))
except Exception as e:
print(e)
for jpgfile in glob.glob("F:\ChengXuCode\pytorch-yolo-v3-master\imgs\*.jpg"):
# print(jpgfile)
convertjpg(jpgfile,"F:\ChengXuCode\pytorch-yolo-v3-master\pro_img")
注意:glob安装时要用
#如果是Python3,则用
pip install glob3
#如果是Python2
pip install glob2