Convert CRLF to LF
This commit is contained in:
chun-awa
parent
cb21d05b9a
commit
d549d7a160
272
auto_pack_img.py
272
auto_pack_img.py
@ -1,136 +1,136 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
import random
|
import random
|
||||||
|
|
||||||
import cv2
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
|
||||||
class Node:
|
class Node:
|
||||||
def __init__(self, x, y, width, height):
|
def __init__(self, x, y, width, height):
|
||||||
self.x = x
|
self.x = x
|
||||||
self.y = y
|
self.y = y
|
||||||
self.width = width
|
self.width = width
|
||||||
self.height = height
|
self.height = height
|
||||||
self.used = False
|
self.used = False
|
||||||
self.right = None
|
self.right = None
|
||||||
self.down = None
|
self.down = None
|
||||||
|
|
||||||
|
|
||||||
def find_node(root, width, height):
|
def find_node(root, width, height):
|
||||||
if root.used:
|
if root.used:
|
||||||
return find_node(root.right, width, height) or find_node(root.down, width, height)
|
return find_node(root.right, width, height) or find_node(root.down, width, height)
|
||||||
elif width <= root.width and height <= root.height:
|
elif width <= root.width and height <= root.height:
|
||||||
return root
|
return root
|
||||||
else:
|
else:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
def split_node(node, width, height):
|
def split_node(node, width, height):
|
||||||
node.used = True
|
node.used = True
|
||||||
node.down = Node(node.x, node.y + height, node.width, node.height - height)
|
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)
|
node.right = Node(node.x + width, node.y, node.width - width, height)
|
||||||
return node
|
return node
|
||||||
|
|
||||||
|
|
||||||
def grow_node(root, width, height):
|
def grow_node(root, width, height):
|
||||||
can_grow_down = (width <= root.width)
|
can_grow_down = (width <= root.width)
|
||||||
can_grow_right = (height <= root.height)
|
can_grow_right = (height <= root.height)
|
||||||
|
|
||||||
should_grow_right = can_grow_right and (root.height >= (root.width + width))
|
should_grow_right = can_grow_right and (root.height >= (root.width + width))
|
||||||
should_grow_down = can_grow_down and (root.width >= (root.height + height))
|
should_grow_down = can_grow_down and (root.width >= (root.height + height))
|
||||||
|
|
||||||
if should_grow_right:
|
if should_grow_right:
|
||||||
return grow_right(root, width, height)
|
return grow_right(root, width, height)
|
||||||
elif should_grow_down:
|
elif should_grow_down:
|
||||||
return grow_down(root, width, height)
|
return grow_down(root, width, height)
|
||||||
elif can_grow_right:
|
elif can_grow_right:
|
||||||
return grow_right(root, width, height)
|
return grow_right(root, width, height)
|
||||||
elif can_grow_down:
|
elif can_grow_down:
|
||||||
return grow_down(root, width, height)
|
return grow_down(root, width, height)
|
||||||
else:
|
else:
|
||||||
root.width=root.width+width//2
|
root.width=root.width+width//2
|
||||||
root.height=root.height+height//2
|
root.height=root.height+height//2
|
||||||
return grow_node(root, width, height)
|
return grow_node(root, width, height)
|
||||||
|
|
||||||
|
|
||||||
def grow_right(root, width, height):
|
def grow_right(root, width, height):
|
||||||
new_root = Node(0, 0, root.width + width, root.height)
|
new_root = Node(0, 0, root.width + width, root.height)
|
||||||
new_root.used = True
|
new_root.used = True
|
||||||
new_root.down = root
|
new_root.down = root
|
||||||
new_root.right = Node(root.width, 0, width, root.height)
|
new_root.right = Node(root.width, 0, width, root.height)
|
||||||
return new_root if find_node(new_root, width, height) else None
|
return new_root if find_node(new_root, width, height) else None
|
||||||
|
|
||||||
|
|
||||||
def grow_down(root, width, height):
|
def grow_down(root, width, height):
|
||||||
new_root = Node(0, 0, root.width, root.height + height)
|
new_root = Node(0, 0, root.width, root.height + height)
|
||||||
new_root.used = True
|
new_root.used = True
|
||||||
new_root.right = root
|
new_root.right = root
|
||||||
new_root.down = Node(0, root.height, root.width, height)
|
new_root.down = Node(0, root.height, root.width, height)
|
||||||
return new_root if find_node(new_root, width, height) else None
|
return new_root if find_node(new_root, width, height) else None
|
||||||
|
|
||||||
|
|
||||||
def stitch_images_bin_packing(images):
|
def stitch_images_bin_packing(images):
|
||||||
|
|
||||||
root = Node(0, 0, images[0].shape[1], images[0].shape[0])
|
root = Node(0, 0, images[0].shape[1], images[0].shape[0])
|
||||||
|
|
||||||
positions = []
|
positions = []
|
||||||
|
|
||||||
for img in images:
|
for img in images:
|
||||||
node = find_node(root, img.shape[1], img.shape[0])
|
node = find_node(root, img.shape[1], img.shape[0])
|
||||||
if node:
|
if node:
|
||||||
split_node(node, img.shape[1], img.shape[0])
|
split_node(node, img.shape[1], img.shape[0])
|
||||||
else:
|
else:
|
||||||
root = grow_node(root, img.shape[1], img.shape[0])
|
root = grow_node(root, img.shape[1], img.shape[0])
|
||||||
node = find_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])
|
split_node(node, img.shape[1], img.shape[0])
|
||||||
positions.append((node.x, node.y))
|
positions.append((node.x, node.y))
|
||||||
|
|
||||||
|
|
||||||
# stitched_image = np.zeros((root.height, root.width, 3), dtype=np.uint8)
|
# 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)
|
stitched_image = np.full((root.height, root.width, 3),255, dtype=np.uint8)
|
||||||
|
|
||||||
|
|
||||||
for pos, img in zip(positions, images):
|
for pos, img in zip(positions, images):
|
||||||
x, y = pos
|
x, y = pos
|
||||||
stitched_image[y:y + img.shape[0], x:x + img.shape[1]] = img
|
stitched_image[y:y + img.shape[0], x:x + img.shape[1]] = img
|
||||||
|
|
||||||
return stitched_image
|
return stitched_image
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def load_images_from_folder(folder,max_height=768,max_width=768):
|
def load_images_from_folder(folder,max_height=768,max_width=768):
|
||||||
images = []
|
images = []
|
||||||
for filename in os.listdir(folder):
|
for filename in os.listdir(folder):
|
||||||
img = cv2.imread(os.path.join(folder, filename))
|
img = cv2.imread(os.path.join(folder, filename))
|
||||||
if img is not None:
|
if img is not None:
|
||||||
if max_height is not None and max_width 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:
|
if img.shape[0] > max_height or img.shape[1] > max_width:
|
||||||
scw=img.shape[1]/max_width
|
scw=img.shape[1]/max_width
|
||||||
sch=img.shape[0]/max_height
|
sch=img.shape[0]/max_height
|
||||||
mac=max(scw,sch)
|
mac=max(scw,sch)
|
||||||
img = cv2.resize(img, (int(img.shape[1]//mac),int(img.shape[0]//mac)))
|
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)))
|
||||||
# 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)
|
images.append(img)
|
||||||
random.shuffle(images)
|
random.shuffle(images)
|
||||||
return images
|
return images
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
folder = 'zh_cn'
|
folder = 'zh_cn'
|
||||||
images = load_images_from_folder(folder)
|
images = load_images_from_folder(folder)
|
||||||
|
|
||||||
stitched_image = stitch_images_bin_packing(images)
|
stitched_image = stitch_images_bin_packing(images)
|
||||||
|
|
||||||
|
|
||||||
cv2.imwrite('images/auto_img_zh_cn.png', stitched_image)
|
cv2.imwrite('images/auto_img_zh_cn.png', stitched_image)
|
||||||
|
|
||||||
|
|
||||||
folder = 'en'
|
folder = 'en'
|
||||||
images = load_images_from_folder(folder)
|
images = load_images_from_folder(folder)
|
||||||
|
|
||||||
stitched_image = stitch_images_bin_packing(images)
|
stitched_image = stitch_images_bin_packing(images)
|
||||||
|
|
||||||
|
|
||||||
cv2.imwrite('images/auto_img_en.png', stitched_image)
|
cv2.imwrite('images/auto_img_en.png', stitched_image)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user