将png图片转化为RGB三通道

先看看自己pytorch的一些设置是否正确

 

代码

import os
from PIL import Image
# 原文件夹名称
folder_path = "Data"
#转化后的文件夹名称
output_folder = "Converted"

# 创建输出文件夹
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 遍历文件夹中的文件
for filename in os.listdir(folder_path):
    file_path = os.path.join(folder_path, filename)
  
    # 检查文件是否是PNG图像
    if filename.lower().endswith(".png"):
        
        # 打开图像并将通道数转换为RGB
        image = Image.open(file_path)
        rgb_image = image.convert("RGB")
        
        # 构造保存路径和文件名
        save_path = os.path.join(output_folder, filename)
        
        # 保存为新的图像文件
        rgb_image.save(save_path)
        
        print(f"{filename} 的通道数已转换为RGB三通道，保存为 {save_path}")