只需要修改注释对应地方
import os
from PIL import Image
# size为需要压缩后的尺寸大小
def resize_images(input_folder, output_folder, size=(640, 640)):
if not os.path.exists(output_folder):
os.makedirs(output_folder)
for filename in os.listdir(input_folder):
if filename.lower().endswith(('.jpg', '.png', '.jpeg')):
img_path = os.path.join(input_folder, filename)
img = Image.open(img_path)
img_resized = img.resize(size, Image.LANCZOS)
img_resized.save(os.path.join(output_folder, filename))
print(f'Resized and saved: {filename}')
input_folder = 'ys_data/enhanced_images' # 输入图片路径
output_folder = 'ys_data/enhanced_ys_images' # 输出图片路径
resize_images(input_folder, output_folder)