工具——批处理图像缩放指定大小

之前做地平线智能车的时候，使用了resnet巡线，由于resnet只能训练224*224的图像，所以写了这个批处理脚本

import os
import cv2


def resize_image(input_image_path, output_image_path, width, height):
    # 读取图像
    image = cv2.imread(input_image_path)

    if image is None:
        print(f"无法读取图像: {input_image_path}")
        return

    # 调整图像大小
    resized_image = cv2.resize(image, (width, height))

    # 保存调整大小后的图像
    cv2.imwrite(output_image_path, resized_image)
    print(f"图像已保存到 {output_image_path}")


def batch_resize_images(input_folder, output_folder, new_width, new_height):
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 遍历输入文件夹中的所有文件
    for filename in os.listdir(input_folder):
        if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
            input_image_path = os.path.join(input_folder, filename)
            output_image_path = os.path.join(output_folder, filename)
            resize_image(input_image_path, output_image_path, new_width, new_height)


# 示例使用
input_folder = '/home/moon/sunrise/src/2'  # 输入文件夹路径
output_folder = '/home/moon/sunrise/src/biaoding'  # 输出文件夹路径
new_width = 224
new_height = 224

batch_resize_images(input_folder, output_folder, new_width, new_height)