一些基本图像预处理---将一个文件中的所有图片进行尺度归一化、查看每个像素的RGB值，将指定颜色切换为白色、将文件中所有图片从png到jpg的切换

目录

1.将一个文件中的所有图片进行尺度归一化

2. 查看一张图片的每个像素的RGB值

3.将指定颜色切换成其他颜色（代码中将红色切换为白色）

 4.将一个文件中所有图片从png到jpg的切换

5.将一个文件夹中的所有图像重命名从1开始 

1.将一个文件中的所有图片进行尺度归一化

实例中，本代码是将图像全部归一为352*352，可以调整参数，也需要将文件地址切换所要处理的文件地址。

from PIL import Image
import os


def resize_images(folder_path, output_size):
    # 获取文件夹中的所有文件
    file_list = os.listdir(folder_path)

    for file_name in file_list:
        # 检查文件是否为图片文件（可根据需要修改条件）
        if file_name.endswith(('.jpg', '.jpeg', '.png')):
            file_path = os.path.join(folder_path, file_name)

            # 打开并调整图片大小
            image = Image.open(file_path)
            resized_image = image.resize(output_size)

            # 保存调整大小后的图片
            resized_image.save(file_path)

            print(f"Resized image: {file_name}")


# 指定要处理的文件夹路径和输出大小
folder_path = './TrainDate/masks'  # 替换为实际的文件夹路径
output_size = (352, 352)  # 替换为想要的输出大小

# 调用函数进行图片大小归一化
resize_images(folder_path, output_size)

2. 查看一张图片的每个像素的RGB值

from PIL import Image

def get_image_rgb_values(image_path):
    # 打开图片
    image = Image.open(image_path)

    # 将图片转换为RGB模式（忽略透明度）
    image = image.convert('RGB')

    # 获取图片的像素数据
    pixels = image.load()
    width, height = image.size

    # 遍历每个像素
    for x in range(width):
        for y in range(height):
            r, g, b = pixels[x, y]
            # 输出RGB值
            print(f"Pixel at ({x}, {y}): R={r}, G={g}, B={b}")

    # 关闭图片
    image.close()

# 指定图片路径
image_path = '1.png'

# 调用函数查看图片RGB值
get_image_rgb_values(image_path)

3.将指定颜色切换成其他颜色（代码中将红色切换为白色）

换成指定颜色需要修改 以下部分代码参数：

 # 判断是否为红色区域（阈值可根据需要进行调整）
                if r > 100 and g >100 and b <100:
                    # 将红色区域替换为白色0
                    pixels[x, y] = (255, 255, 255, a)

from PIL import Image
import os

def replace_red_with_white(folder_path):
    # 获取文件夹中的所有文件名
    filenames = os.listdir(folder_path)
    # 选取所有PNG图片文件
    image_files = [filename for filename in filenames if filename.endswith('.png')]

    # 遍历每个图片文件并进行处理
    for image_file in image_files:
        image_path = os.path.join(folder_path, image_file)
        # 打开图片
        image = Image.open(image_path)

        # 将图片转换为RGBA模式
        image = image.convert('RGBA')

        # 获取图片的像素数据
        pixels = image.load()
        width, height = image.size

        # 遍历每个像素
        for x in range(width):
            for y in range(height):
                r, g, b, a = pixels[x, y]

                # 判断是否为红色区域（阈值可根据需要进行调整）
                if r > 100 and g >100 and b <100:
                    # 将红色区域替换为白色0
                    pixels[x, y] = (255, 255, 255, a)

        # 保存修改后的图片
        new_image_path = os.path.join(folder_path, image_file)
        image.save(new_image_path)

        # 关闭图片
        image.close()

        print(f"Processed {image_file} and saved as {os.path.basename(new_image_path)}")

# 指定文件夹路径
folder_path = './TrainData/masks'

# 调用函数进行处理
replace_red_with_white(folder_path)

效果图如下所示： 

 

 4.将一个文件中所有图片从png到jpg的切换

from PIL import Image
import os

def convert_png_to_jpg(input_folder, output_folder):
    # 遍历输入文件夹中的所有文件
    for filename in os.listdir(input_folder):
        if filename.endswith(".png"):
            # 构建输入和输出文件的路径
            input_path = os.path.join(input_folder, filename)
            output_filename = os.path.splitext(filename)[0] + ".jpg"
            output_path = os.path.join(output_folder, output_filename)

            # 打开PNG图像并保存为JPG格式
            image = Image.open(input_path)
            image.convert("RGB").save(output_path, "JPEG")
            print(f"Converted {input_path} to {output_path}")

# 指定输入文件夹和输出文件夹的路径
input_folder = "input_folder"
output_folder = "output_folder"

# 调用函数进行转换
convert_png_to_jpg(input_folder, output_folder)

5.将一个文件夹中的所有图像重命名从1开始 

import os

def rename_images(folder_path):
    # 获取目标文件夹中的所有文件
    file_list = os.listdir(folder_path)
    image_extensions = ['.jpg', '.jpeg', '.png', '.gif']  # 支持的图片文件扩展名

    # 过滤出图片文件
    image_files = [file for file in file_list if os.path.splitext(file)[1].lower() in image_extensions]

    # 重命名图片文件
    for i, image_file in enumerate(image_files):
        extension = os.path.splitext(image_file)[1]
        new_name = str(i + 1) + extension
        old_path = os.path.join(folder_path, image_file)
        new_path = os.path.join(folder_path, new_name)
        os.rename(old_path, new_path)
        print(f"{image_file} 重命名为 {new_name}")

# 调用示例
folder_path = "./TrainDataset/image"
rename_images(folder_path)