126 lines
4.5 KiB
Python
Executable File
126 lines
4.5 KiB
Python
Executable File
import base64
|
||
import io
|
||
import os
|
||
from PIL import Image, ImageColor
|
||
|
||
|
||
class ImageProCls:
|
||
base64FilePath=None
|
||
|
||
@staticmethod
|
||
def humanbody_blending_with_image(input_image, gray_image, bg_image):
|
||
"""
|
||
|
||
:param input_image: the PIL.Image instance, the source humanbody image
|
||
:param gray_image: the PIL.Image instance, it is created from api base64 result, a gray image
|
||
:param bg_image: the PIL.Image instance, the background image you want to replace
|
||
:return: the PIL.Image instance is blending with humanbody
|
||
|
||
notes: you should close the return object after you leave
|
||
"""
|
||
input_width, input_height = input_image.size
|
||
bg_width, bg_height = bg_image.size
|
||
|
||
input_aspect_ratio = input_width / float(input_height)
|
||
bg_aspect_ratio = bg_width / float(bg_height)
|
||
|
||
if bg_aspect_ratio > input_aspect_ratio:
|
||
target_width, target_height = int(bg_height * input_aspect_ratio), bg_height
|
||
else:
|
||
target_width, target_height = bg_width, int(bg_width / input_aspect_ratio)
|
||
|
||
crop_image = bg_image.crop((0, 0, 0+target_width, 0+target_height))
|
||
new_image = crop_image.resize((input_width, input_height))
|
||
crop_image.close()
|
||
|
||
for x in range(0, input_width):
|
||
for y in range(0, input_height):
|
||
coord = (x, y)
|
||
gray_pixel_value = gray_image.getpixel(coord)
|
||
input_rgb_color = input_image.getpixel(coord)
|
||
bg_rgb_color = new_image.getpixel(coord)
|
||
|
||
confidence = gray_pixel_value / 255.0
|
||
alpha = confidence
|
||
|
||
R = input_rgb_color[0] * alpha + bg_rgb_color[0] * (1 - alpha)
|
||
G = input_rgb_color[1] * alpha + bg_rgb_color[1] * (1 - alpha)
|
||
B = input_rgb_color[2] * alpha + bg_rgb_color[2] * (1 - alpha)
|
||
|
||
R = max(0, min(int(R), 255))
|
||
G = max(0, min(int(G), 255))
|
||
B = max(0, min(int(B), 255))
|
||
|
||
new_image.putpixel(coord, (R, G, B))
|
||
|
||
return new_image
|
||
|
||
@staticmethod
|
||
def humanbody_blending_with_color(input_image, gray_image, bg_color):
|
||
"""
|
||
:param input_image: the PIL.Image instance
|
||
:param gray_image: the PIL.Image instance, it is created from api base64 result, it is a gray image
|
||
:param bg_color: a color string value, such as '#FFFFFF'
|
||
:return: PIL.Image instance
|
||
|
||
notes: you should close the return object after you leave
|
||
"""
|
||
input_width, input_height = input_image.size
|
||
bg_rgb_color = ImageColor.getrgb(bg_color)
|
||
|
||
new_image = Image.new("RGB", input_image.size, bg_color)
|
||
|
||
for x in range(0, input_width):
|
||
for y in range(0, input_height):
|
||
coord = (x, y)
|
||
gray_pixel_value = gray_image.getpixel(coord)
|
||
input_rgb_color = input_image.getpixel(coord)
|
||
|
||
confidence = gray_pixel_value / 255.0
|
||
alpha = confidence
|
||
|
||
R = input_rgb_color[0] * alpha + bg_rgb_color[0] * (1 - alpha)
|
||
G = input_rgb_color[1] * alpha + bg_rgb_color[1] * (1 - alpha)
|
||
B = input_rgb_color[2] * alpha + bg_rgb_color[2] * (1 - alpha)
|
||
|
||
R = max(0, min(int(R), 255))
|
||
G = max(0, min(int(G), 255))
|
||
B = max(0, min(int(B), 255))
|
||
|
||
new_image.putpixel(coord, (R, G, B))
|
||
|
||
return new_image
|
||
|
||
@staticmethod
|
||
def getSegmentImg(filePath):
|
||
input_file = ''
|
||
with open(filePath, 'r') as f:
|
||
input_file = f.read()
|
||
input_file = base64.b64decode(input_file)
|
||
|
||
gray_image = Image.open(io.BytesIO(input_file))
|
||
input_image = Image.open('./imgResource/segment.jpg', 'r')
|
||
|
||
new_image = ImageProCls.humanbody_blending_with_color(input_image, gray_image, '#FFFFFF')
|
||
new_image.save('./imgResource/resultImg.jpg')
|
||
print('-' * 60)
|
||
print('结果已经生成，生成文件名：resultImg.jpg，请在imgResource/目录下查看')
|
||
if os.path.exists(filePath):
|
||
os.remove(filePath)
|
||
f.close()
|
||
new_image.close()
|
||
input_image.close()
|
||
gray_image.close()
|
||
|
||
@staticmethod
|
||
def getMergeImg(base64Str):
|
||
imgdata = base64.b64decode(base64Str)
|
||
file = open('./imgResource/MergeResultImg.jpg', 'wb')
|
||
file.write(imgdata)
|
||
file.close()
|
||
print('结果已经生成，生成文件名：MergeResultImg.jpg，请在imgResource/目录下查看')
|
||
|
||
|
||
|
||
|