python批量修改图片尺寸

在图片识别项目中遇到的数据集尺寸修改问题，需要将测试集的一千多张图片修改为100X128尺寸，同时名字和顺序不改变，并全部输出为png格式。
代码即注释如下：

import os
import time
from PIL import Image

# 批量修改图片尺寸，输入原图目录和修改后存放目录地址即可
def resize_batch(path, save_path):
    files = os.listdir(path)  # 返回path目录下的所有文件名
    
    # 遍历每一张图片并修改其尺寸
    for i in files:
      f = 0
      document = os.path.join(path, i)  # 返回path和i拼接之后的路径，即第i张图片
      img = Image.open(document)  # 读取第i张图片
      img_resize = img.resize((100, 128))  # 修改为100x128尺寸
      fileName = str(i)[:-4]  # 原图除后缀外的名字，这里原图后缀是.jpg
      img_resize.save(save_path + os.sep + '%s.png' % fileName)  # 保存路径，其中os.sep为系统分隔符