first commit

app/lib/face/ImagePro.py

@@ -0,0 +1,125 @@
+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/目录下查看')
+
+
+
+