241015_把文件夹中所有的图片缩放为指定大小(缩放数据集)
在制作自己数据集的过程中,我们经常会遇到image尺寸不统一的情况,以下是一段缩放到指定size的代码(代码中以512*512举例),修改路径即可运行
from PIL import Image
import osdef resize_images(source_folder, target_folder):# 确保目标文件夹存在if not os.path.exists(target_folder):os.makedirs(target_folder)# 遍历源文件夹中的所有文件for filename in os.listdir(source_folder):if filename.lower().endswith(('.png', '.jpg', '.jpeg')):# 构建完整的文件路径file_path = os.path.join(source_folder, filename)# 打开图像文件with Image.open(file_path) as img:# 缩放图像到512x512大小resized_img = img.resize((512, 512))# 新的文件路径new_file_path = os.path.join(target_folder, filename)# 保存缩放后的图像resized_img.save(new_file_path)# 使用示例
source_folder = 'D:\Softwares\Codes\Git\GitProgram\light\data\\atm_origin' # 源文件夹路径
target_folder = 'D:\Softwares\Codes\Git\GitProgram\light\data\\atm_512_test' # 目标文件夹路径
resize_images(source_folder, target_folder)