96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
import json
|
|
from PIL import Image
|
|
import os
|
|
|
|
# 全局变量定义分辨率
|
|
RESIZE_TO = (512, 512)
|
|
LARGE_IMAGE_SIZE = (4096, 2048)
|
|
IMAGE_COUNT = 32
|
|
|
|
ROOT_DIR = '../example/src/main/assets/pic'
|
|
|
|
def load_json(file_path):
|
|
with open(file_path, 'r') as file:
|
|
return json.load(file)
|
|
|
|
def remove_white_border(image):
|
|
"""
|
|
去除图片边缘的一个像素宽的白色边框。
|
|
:param image: 输入的PIL Image对象。
|
|
:return: 处理后的PIL Image对象。
|
|
"""
|
|
# 获取图像尺寸
|
|
width, height = image.size
|
|
|
|
# 定义白色阈值，允许一定的误差范围
|
|
white_threshold = (250, 250, 250)
|
|
|
|
# 查找左边界
|
|
left = 0
|
|
while left < width and all(image.getpixel((left, y))[:3] >= white_threshold for y in range(height)):
|
|
left += 1
|
|
|
|
# 查找右边界
|
|
right = width - 1
|
|
while right > 0 and all(image.getpixel((right, y))[:3] >= white_threshold for y in range(height)):
|
|
right -= 1
|
|
|
|
# 查找上边界
|
|
top = 0
|
|
while top < height and all(image.getpixel((x, top))[:3] >= white_threshold for x in range(width)):
|
|
top += 1
|
|
|
|
# 查找下边界
|
|
bottom = height - 1
|
|
while bottom > 0 and all(image.getpixel((x, bottom))[:3] >= white_threshold for x in range(width)):
|
|
bottom -= 1
|
|
|
|
# 如果发现有白色边框，则裁剪它们
|
|
if left < right and top < bottom:
|
|
image = image.crop((left, top, right + 1, bottom + 1))
|
|
|
|
return image
|
|
|
|
def resize_image(image_path):
|
|
image = Image.open(image_path).convert('RGBA')
|
|
image = remove_white_border(image)
|
|
#return image.resize(RESIZE_TO)
|
|
return image
|
|
|
|
def main(json_file_path):
|
|
# 确保使用完整路径
|
|
json_file_path = os.path.join(ROOT_DIR, json_file_path)
|
|
data = load_json(json_file_path)
|
|
|
|
new_data = {}
|
|
|
|
for folder_name, images in data.items():
|
|
large_image = Image.new('RGBA', LARGE_IMAGE_SIZE, color=(0, 0, 0, 0))
|
|
images_coordinates = {}
|
|
index = 0
|
|
|
|
for image_name in images:
|
|
if index >= IMAGE_COUNT:
|
|
break
|
|
# 图片位于子文件夹内
|
|
img_path = os.path.join(ROOT_DIR, folder_name, image_name)
|
|
img = resize_image(img_path)
|
|
x = (index % 8) * RESIZE_TO[0]
|
|
y = (index // 8) * RESIZE_TO[1]
|
|
large_image.paste(img, (x, y))
|
|
images_coordinates[image_name] = {'x': x, 'y': y}
|
|
index += 1
|
|
|
|
output_image_path = os.path.join(ROOT_DIR, f"{folder_name}ex.png")
|
|
large_image.save(output_image_path, format='png')
|
|
|
|
# 将坐标信息直接保存到new_data字典中
|
|
new_data[folder_name] = images_coordinates
|
|
|
|
# 输出新的json文件
|
|
output_json_path = os.path.join(ROOT_DIR, 'motion_data_ex.json')
|
|
with open(output_json_path, 'w') as outfile:
|
|
json.dump(new_data, outfile, indent=4)
|
|
|
|
# 使用示例
|
|
main('motion_data.json') |