Python 将 PNG 图片展平成 RGB 数值并保存
要将文件夹中的多张 PNG 图片展平成 RGB 数值表示,并储存在另一个文件夹中,可以使用 Python 中的 PIL 库来实现。以下是一个示例代码:
from PIL import Image
import os
# 定义输入和输出文件夹路径
input_folder = 'C:/Users/jh/Desktop/data/images_block'
output_folder = 'C:/Users/jh/Desktop/data/images_flatten'
# 遍历输入文件夹中的所有 png 图片
for filename in os.listdir(input_folder):
if filename.endswith('.png'):
# 打开图片
img = Image.open(os.path.join(input_folder, filename))
# 将图片展平为一维数组
flattened_img = img.convert('RGB').flatten()
# 创建输出文件夹(如果不存在)
os.makedirs(output_folder, exist_ok=True)
# 保存展平后的图片
flattened_img.save(os.path.join(output_folder, filename))
在上述代码中,首先使用 os.listdir 函数遍历输入文件夹中的所有文件,然后通过判断文件扩展名是否为 '.png' 筛选出 PNG 图片。接下来,使用 Image.open 函数打开每个图片,然后使用 convert('RGB') 方法将图像转换为 RGB 模式,最后使用 flatten 方法将图像展平为一维数组。然后,使用 os.makedirs 函数创建输出文件夹(如果不存在),最后使用 save 方法将展平后的图片保存到输出文件夹中。
原文地址: https://www.cveoy.top/t/topic/fWgR 著作权归作者所有。请勿转载和采集!