python测试模型图像处理性能

from PIL import Image
import time
import os  # 导入 os 模块

# 读取并处理图像的函数
def process_image(file_path, size):
    start_time = time.time()
    image = Image.open(file_path)
    processed_image = image.resize(size)  # 调整图像大小
    output_dir = 'processed_images'
    if not os.path.exists(output_dir):  # 如果目录不存在则创建
        os.makedirs(output_dir)
    output_path = os.path.join(output_dir, 'processed_' + os.path.basename(file_path))
    processed_image.save(output_path)  # 将处理后的图像保存到文件
    end_time = time.time()
    return end_time - start_time  # 返回处理时间



# 测试不同大小和格式的图像文件
image_files = ['xx/xx.jpg', 'xx/xx.png']
sizes = [(200, 200), (500, 500)]  # 不同的尺寸

for file in image_files:
    for size in sizes:
        processing_time = process_image(file, size)
        print(f'Processing time for {file} with size {size}: {processing_time} seconds')