from PIL import Image
import os

input_folder = r'D:\1BNS_projects\yolo_safehat_num_v8\datasets\2m'  #要调整的图片所在的文件夹名称
output_folder = r'D:\1BNS_projects\yolo_safehat_num_v8\datasets\2m1' # 调整后的图片所放的文件夹名称
if not os.path.exists(output_folder):
    os.makedirs(output_folder)
new_size = (640, 640)  # 调整为256*256

for filename in os.listdir(input_folder):
    # Check if the file is an image
    if filename.endswith('.png') or filename.endswith('.jpg'):
        # Open the image and resize it
        img = Image.open(os.path.join(input_folder, filename))
        img = img.resize(new_size)

        # Save the resized image to the output folder
        new_filename = os.path.splitext(filename)[0] + '.jpg'
        img.save(os.path.join(output_folder, new_filename))