nodes/AbyssBadger0-ComfyUI_BadgerTools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

像素函数结构优化

Browse Source
This commit is contained in:
changjiaxiang
2024-07-30 18:13:20 +08:00
parent 0809b598cf
commit 9eb6b25e2a
2 changed files with 32 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+1 -32
View File
@@ -1508,41 +1508,10 @@ class ToPixel:
FUNCTION = "image_to_pixel"
def image_to_pixel(self, original_image, threshold, pix, tile_size, color_card=None):
regular_size = tile_size*pix
original_image = tensorToImg(original_image)
original_pixels = regular_image(original_image,output_size=(regular_size,regular_size))
# 创建新的图像，用于存储像素化的结果
pixelated_image = Image.new('RGB', (pix, pix))
if color_card!=None:
color_card = tensorToImg(color_card)
color_palette = load_color_card(color_card)
# 遍历每个8x8的方块
for i in range(0, regular_size, tile_size):
for j in range(0, regular_size, tile_size):
# 获取当前方块
block = original_pixels[i:i+tile_size, j:j+tile_size].reshape(-1, 3)
# 找到主要颜色
dominant_color = find_dominant_color(block, threshold)
# 匹配到颜色卡中的颜色
matched_color = match_color_to_palette(dominant_color, color_palette)
# 将匹配的颜色赋给对应的像素点
pixelated_image.putpixel((j // tile_size, i // tile_size), matched_color)
else:
# 遍历每个8x8的方块
for i in range(0, regular_size, tile_size):
for j in range(0, regular_size, tile_size):
# 获取当前方块
block = original_pixels[i:i+tile_size, j:j+tile_size].reshape(-1, 3)
# 找到主要颜色
dominant_color = find_dominant_color(block, threshold)
# 将主要颜色的平均值赋给对应的像素点
pixelated_image.putpixel((j // tile_size, i // tile_size), tuple(dominant_color.astype(int)))
pixelated_image = to_pixel(original_image, threshold, pix, tile_size, color_card)
pixelated_image = imgToTensor(pixelated_image)
garbage_collect()
return (pixelated_image,)
pixel.py
+31 -1
View File
@@ -58,4 +58,34 @@ def regular_image(original_image, output_size=(512, 512), fill_color=(255, 255,
new_img.paste(original_image, (left, top))
original_pixels = np.array(new_img)
return original_pixels
return original_pixels
def to_pixel(original_image, threshold, pix, tile_size, color_card=None):
regular_size = tile_size*pix
original_pixels = regular_image(original_image,output_size=(regular_size,regular_size))
# 创建新的图像，用于存储像素化的结果
pixelated_image = Image.new('RGB', (pix, pix))
if color_card!=None:
color_palette = load_color_card(color_card)
# 遍历每个8x8的方块
for i in range(0, regular_size, tile_size):
for j in range(0, regular_size, tile_size):
# 获取当前方块
block = original_pixels[i:i+tile_size, j:j+tile_size].reshape(-1, 3)
# 找到主要颜色
dominant_color = find_dominant_color(block, threshold)
# 匹配到颜色卡中的颜色
matched_color = match_color_to_palette(dominant_color, color_palette)
# 将匹配的颜色赋给对应的像素点
pixelated_image.putpixel((j // tile_size, i // tile_size), matched_color)
else:
# 遍历每个8x8的方块
for i in range(0, regular_size, tile_size):
for j in range(0, regular_size, tile_size):
# 获取当前方块
block = original_pixels[i:i+tile_size, j:j+tile_size].reshape(-1, 3)
# 找到主要颜色
dominant_color = find_dominant_color(block, threshold)
# 将主要颜色的平均值赋给对应的像素点
pixelated_image.putpixel((j // tile_size, i // tile_size), tuple(dominant_color.astype(int)))
return pixelated_image