78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import argparse
|
||
from PIL import Image
|
||
|
||
def rgb_to_rg565(r, g, b):
|
||
# 将RGB数据转换为RG565格式
|
||
r5 = (r >> 3) & 0x1F
|
||
g6 = (g >> 2) & 0x3F
|
||
b5 = (b >> 3) & 0x1F
|
||
return (r5 << 11) | (g6 << 5) | b5
|
||
|
||
def rg565_to_rgb(rg565):
|
||
# 将RG565数据转换回RGB数据
|
||
r = (rg565 >> 11) & 0x1F
|
||
g = (rg565 >> 5) & 0x3F
|
||
b = rg565 & 0x1F
|
||
return (r << 3, g << 2, b << 3)
|
||
|
||
def convert_to_rg565(img):
|
||
# 获取图片的像素数据
|
||
pixels = img.getdata()
|
||
|
||
# 将每个像素的RGB数据转换为RG565格式
|
||
# rg565_pixels = [rgb_to_rg565(pixel[0], pixel[1], pixel[2]) for pixel in pixels]
|
||
rg565_pixels = [rgb_to_rg565(pixel[2], pixel[1], pixel[0]) for pixel in pixels]
|
||
# 创建一个新的图片对象
|
||
new_img = Image.new('RGB', img.size)
|
||
|
||
# 将RG565数据转换回RGB数据并设置新的像素数据
|
||
new_pixels = [rg565_to_rgb(pixel) for pixel in rg565_pixels]
|
||
new_img.putdata(new_pixels)
|
||
|
||
return new_img
|
||
|
||
def convert_to_jpeg(input_image_path, output_image_path):
|
||
# 打开输入图片
|
||
with Image.open(input_image_path) as img:
|
||
# 获取原始图片的尺寸
|
||
original_size = img.size
|
||
|
||
# 将图片转换为RGB模式（如果它不是RGB模式）
|
||
if img.mode != 'RGB':
|
||
img = img.convert('RGB')
|
||
|
||
# 将RGB数据转换为RG565格式
|
||
img = convert_to_rg565(img)
|
||
|
||
# 确保图片的尺寸与原始尺寸一致
|
||
if img.size != original_size:
|
||
img = img.resize(original_size)
|
||
|
||
# 保存为JPEG格式
|
||
img.save(output_image_path, 'JPEG')
|
||
|
||
if __name__ == "__main__":
|
||
# 创建命令行参数解析器
|
||
parser = argparse.ArgumentParser(description="Convert a 24-bit color image to a 16-bit color (RG565) image and save as JPEG.")
|
||
|
||
# 添加输入图片路径参数
|
||
parser.add_argument('input_image', type=str, help='Path to the input image file')
|
||
|
||
# 添加输出目录参数
|
||
parser.add_argument('output_directory', type=str, help='Path to the output directory')
|
||
|
||
# 解析命令行参数
|
||
args = parser.parse_args()
|
||
|
||
# 获取输入图片路径
|
||
input_image_path = args.input_image
|
||
|
||
# 获取输出目录路径
|
||
output_directory = args.output_directory
|
||
|
||
# 构建输出图片路径
|
||
output_image_path = f"{output_directory}/{input_image_path.split('/')[-1].split('.')[0]}.jpg"
|
||
|
||
# 调用函数进行转换
|
||
convert_to_jpeg(input_image_path, output_image_path)
|
||
print(f"图片已保存为 {output_image_path}") |