from PIL import Image
import os
import iodef resize_images_in_directory(directory, target_size=(240, 240), max_file_size_kb=500): #保证处理后的图像集 大小为240*240 且不超过 500Kfor filename in os.listdir(directory):if filename.lower().endswith(('.png', '.jpg', '.jpeg')):  # 这里我添加了lower()以确保文件扩展名的大小写不敏感file_path = os.path.join(directory, filename)with Image.open(file_path) as img:if img.mode != 'RGB':img = img.convert('RGB')img = img.resize(target_size, Image.ANTIALIAS)# 初始化保存质量save_quality = 95while True:# 使用 BytesIO 临时保存图像来检查大小buffer = io.BytesIO()img.save(buffer, format='JPEG', quality=save_quality)buffer.seek(0)  # 重置指针到文件的开始size_kb = len(buffer.getvalue()) / 1024  # 获取buffer的大小if size_kb <= max_file_size_kb or save_quality <= 10:breaksave_quality -= 5  # 减少质量以减小文件大小# buffer.close()  # 不需要显示关闭，with语句会处理# 保存调整大小和压缩后的图像with open(file_path, 'wb') as f:f.write(buffer.getvalue())  # 写入文件# 使用函数处理图像
resize_images_in_directory("C:/Users/JJ/Desktop/shousi") # 处理数据集的绝对路径 原始图像集需备份，这里是直接覆盖了