From d549d7a160bca5ba6211b594764c953c3b45596d Mon Sep 17 00:00:00 2001 From: chun-awa <110163919+chun-awa@users.noreply.github.com> Date: Wed, 9 Oct 2024 14:28:17 +0800 Subject: [PATCH] Convert CRLF to LF --- auto_pack_img.py | 272 +++++++++++++++++++++++------------------------ 1 file changed, 136 insertions(+), 136 deletions(-) diff --git a/auto_pack_img.py b/auto_pack_img.py index 3ab6cd7..fed8abf 100755 --- a/auto_pack_img.py +++ b/auto_pack_img.py @@ -1,136 +1,136 @@ -#!/usr/bin/env python3 -import random - -import cv2 -import numpy as np -import os - - -class Node: - def __init__(self, x, y, width, height): - self.x = x - self.y = y - self.width = width - self.height = height - self.used = False - self.right = None - self.down = None - - -def find_node(root, width, height): - if root.used: - return find_node(root.right, width, height) or find_node(root.down, width, height) - elif width <= root.width and height <= root.height: - return root - else: - return None - - -def split_node(node, width, height): - node.used = True - node.down = Node(node.x, node.y + height, node.width, node.height - height) - node.right = Node(node.x + width, node.y, node.width - width, height) - return node - - -def grow_node(root, width, height): - can_grow_down = (width <= root.width) - can_grow_right = (height <= root.height) - - should_grow_right = can_grow_right and (root.height >= (root.width + width)) - should_grow_down = can_grow_down and (root.width >= (root.height + height)) - - if should_grow_right: - return grow_right(root, width, height) - elif should_grow_down: - return grow_down(root, width, height) - elif can_grow_right: - return grow_right(root, width, height) - elif can_grow_down: - return grow_down(root, width, height) - else: - root.width=root.width+width//2 - root.height=root.height+height//2 - return grow_node(root, width, height) - - -def grow_right(root, width, height): - new_root = Node(0, 0, root.width + width, root.height) - new_root.used = True - new_root.down = root - new_root.right = Node(root.width, 0, width, root.height) - return new_root if find_node(new_root, width, height) else None - - -def grow_down(root, width, height): - new_root = Node(0, 0, root.width, root.height + height) - new_root.used = True - new_root.right = root - new_root.down = Node(0, root.height, root.width, height) - return new_root if find_node(new_root, width, height) else None - - -def stitch_images_bin_packing(images): - - root = Node(0, 0, images[0].shape[1], images[0].shape[0]) - - positions = [] - - for img in images: - node = find_node(root, img.shape[1], img.shape[0]) - if node: - split_node(node, img.shape[1], img.shape[0]) - else: - root = grow_node(root, img.shape[1], img.shape[0]) - node = find_node(root, img.shape[1], img.shape[0]) - split_node(node, img.shape[1], img.shape[0]) - positions.append((node.x, node.y)) - - - # stitched_image = np.zeros((root.height, root.width, 3), dtype=np.uint8) - stitched_image = np.full((root.height, root.width, 3),255, dtype=np.uint8) - - - for pos, img in zip(positions, images): - x, y = pos - stitched_image[y:y + img.shape[0], x:x + img.shape[1]] = img - - return stitched_image - - - -def load_images_from_folder(folder,max_height=768,max_width=768): - images = [] - for filename in os.listdir(folder): - img = cv2.imread(os.path.join(folder, filename)) - if img is not None: - if max_height is not None and max_width is not None: - if img.shape[0] > max_height or img.shape[1] > max_width: - scw=img.shape[1]/max_width - sch=img.shape[0]/max_height - mac=max(scw,sch) - img = cv2.resize(img, (int(img.shape[1]//mac),int(img.shape[0]//mac))) - # img = cv2.resize(img, (min(img.shape[1], max_width), min(img.shape[0], max_height))) - # img = cv2.resize(img, (min(img.shape[1], max_width), min(img.shape[0], max_height))) - images.append(img) - random.shuffle(images) - return images - - - -folder = 'zh_cn' -images = load_images_from_folder(folder) - -stitched_image = stitch_images_bin_packing(images) - - -cv2.imwrite('images/auto_img_zh_cn.png', stitched_image) - - -folder = 'en' -images = load_images_from_folder(folder) - -stitched_image = stitch_images_bin_packing(images) - - -cv2.imwrite('images/auto_img_en.png', stitched_image) +#!/usr/bin/env python3 +import random + +import cv2 +import numpy as np +import os + + +class Node: + def __init__(self, x, y, width, height): + self.x = x + self.y = y + self.width = width + self.height = height + self.used = False + self.right = None + self.down = None + + +def find_node(root, width, height): + if root.used: + return find_node(root.right, width, height) or find_node(root.down, width, height) + elif width <= root.width and height <= root.height: + return root + else: + return None + + +def split_node(node, width, height): + node.used = True + node.down = Node(node.x, node.y + height, node.width, node.height - height) + node.right = Node(node.x + width, node.y, node.width - width, height) + return node + + +def grow_node(root, width, height): + can_grow_down = (width <= root.width) + can_grow_right = (height <= root.height) + + should_grow_right = can_grow_right and (root.height >= (root.width + width)) + should_grow_down = can_grow_down and (root.width >= (root.height + height)) + + if should_grow_right: + return grow_right(root, width, height) + elif should_grow_down: + return grow_down(root, width, height) + elif can_grow_right: + return grow_right(root, width, height) + elif can_grow_down: + return grow_down(root, width, height) + else: + root.width=root.width+width//2 + root.height=root.height+height//2 + return grow_node(root, width, height) + + +def grow_right(root, width, height): + new_root = Node(0, 0, root.width + width, root.height) + new_root.used = True + new_root.down = root + new_root.right = Node(root.width, 0, width, root.height) + return new_root if find_node(new_root, width, height) else None + + +def grow_down(root, width, height): + new_root = Node(0, 0, root.width, root.height + height) + new_root.used = True + new_root.right = root + new_root.down = Node(0, root.height, root.width, height) + return new_root if find_node(new_root, width, height) else None + + +def stitch_images_bin_packing(images): + + root = Node(0, 0, images[0].shape[1], images[0].shape[0]) + + positions = [] + + for img in images: + node = find_node(root, img.shape[1], img.shape[0]) + if node: + split_node(node, img.shape[1], img.shape[0]) + else: + root = grow_node(root, img.shape[1], img.shape[0]) + node = find_node(root, img.shape[1], img.shape[0]) + split_node(node, img.shape[1], img.shape[0]) + positions.append((node.x, node.y)) + + + # stitched_image = np.zeros((root.height, root.width, 3), dtype=np.uint8) + stitched_image = np.full((root.height, root.width, 3),255, dtype=np.uint8) + + + for pos, img in zip(positions, images): + x, y = pos + stitched_image[y:y + img.shape[0], x:x + img.shape[1]] = img + + return stitched_image + + + +def load_images_from_folder(folder,max_height=768,max_width=768): + images = [] + for filename in os.listdir(folder): + img = cv2.imread(os.path.join(folder, filename)) + if img is not None: + if max_height is not None and max_width is not None: + if img.shape[0] > max_height or img.shape[1] > max_width: + scw=img.shape[1]/max_width + sch=img.shape[0]/max_height + mac=max(scw,sch) + img = cv2.resize(img, (int(img.shape[1]//mac),int(img.shape[0]//mac))) + # img = cv2.resize(img, (min(img.shape[1], max_width), min(img.shape[0], max_height))) + # img = cv2.resize(img, (min(img.shape[1], max_width), min(img.shape[0], max_height))) + images.append(img) + random.shuffle(images) + return images + + + +folder = 'zh_cn' +images = load_images_from_folder(folder) + +stitched_image = stitch_images_bin_packing(images) + + +cv2.imwrite('images/auto_img_zh_cn.png', stitched_image) + + +folder = 'en' +images = load_images_from_folder(folder) + +stitched_image = stitch_images_bin_packing(images) + + +cv2.imwrite('images/auto_img_en.png', stitched_image)