update

demo/locales.py

@@ -685,4 +685,18 @@ LOCALES = {
             "choices": ["시작"],
         },
     },
+    "template_photo": {
+        "en": {
+            "label": "Social Media Template Photo",
+        },
+        "zh": {
+            "label": "社交媒体模版照",
+        },
+        "ja": {
+            "label": "SNS テンプレート写真",
+        },
+        "ko": {
+            "label": "SNS 템플릿 사진",
+        },
+    },
 }

demo/processor.py

@@ -12,6 +12,7 @@ from hivision.creator.layout_calculator import (
     generate_layout_image,
 )
 from hivision.creator.choose_handler import choose_handler
+from hivision.plugin.template.template_calculator import generte_template_photo
 from demo.utils import range_check
 import gradio as gr
 import os
@@ -141,6 +142,7 @@ class IDPhotoProcessor:
             watermark_text_color,
         )
 
+    # 初始化idphoto_json字典
     def _initialize_idphoto_json(
         self,
         mode_option,
@@ -158,6 +160,7 @@ class IDPhotoProcessor:
             "custom_image_dpi": None,
         }
 
+    # 处理尺寸模式
     def _process_size_mode(
         self,
         idphoto_json,
@@ -202,6 +205,7 @@ class IDPhotoProcessor:
         else:
             idphoto_json["size"] = (None, None)
 
+    # 处理颜色模式
     def _process_color_mode(
         self,
         idphoto_json,
@@ -231,13 +235,16 @@ class IDPhotoProcessor:
                     int(hex_color[i : i + 2], 16) for i in (0, 2, 4)
                 )
             else:
-                raise ValueError("Invalid hex color. You can only use 6 or 7 characters. For example: #FFFFFF or FFFFFF")
+                raise ValueError(
+                    "Invalid hex color. You can only use 6 or 7 characters. For example: #FFFFFF or FFFFFF"
+                )
         else:
             hex_color = LOCALES["bg_color"][language]["develop"][color_option]
             idphoto_json["color_bgr"] = tuple(
                 int(hex_color[i : i + 2], 16) for i in (0, 2, 4)
             )
 
+    # 生成证件照
     def _generate_id_photo(
         self,
         creator: IDCreator,
@@ -272,6 +279,7 @@ class IDPhotoProcessor:
             face_alignment=face_alignment_option,
         )
 
+    # 处理照片生成错误
     def _handle_photo_generation_error(self, language):
         """处理照片生成错误"""
         return [gr.update(value=None) for _ in range(4)] + [
@@ -282,6 +290,7 @@ class IDPhotoProcessor:
             None,
         ]
 
+    # 处理生成的照片
     def _process_generated_photo(
         self,
         result,
@@ -305,20 +314,6 @@ class IDPhotoProcessor:
             result_image_standard, result_image_hd, idphoto_json
         )
 
-        # 生成排版照片
-        result_image_layout, result_image_layout_visible = self._generate_image_layout(
-            idphoto_json,
-            result_image_standard,
-            language,
-            watermark_option,
-            watermark_text,
-            watermark_text_size,
-            watermark_text_opacity,
-            watermark_text_angle,
-            watermark_text_space,
-            watermark_text_color,
-        )
-
         # 添加水印
         if watermark_option == LOCALES["watermark_switch"][language]["choices"][1]:
             result_image_standard, result_image_hd = self._add_watermark(
@@ -331,6 +326,20 @@ class IDPhotoProcessor:
                 watermark_text_space,
                 watermark_text_color,
             )
+        
+        # 生成排版照片
+        result_image_layout, result_image_layout_visible = self._generate_image_layout(
+            idphoto_json,
+            result_image_standard,
+            language,
+        )
+        
+        # 生成模板照片
+        result_image_template, result_image_template_visible = self._generate_image_template(
+            idphoto_json,
+            result_image_hd,
+            language,
+        )
 
         # 调整图片大小
         output_image_path_dict = self._resize_image_if_needed(
@@ -348,6 +357,7 @@ class IDPhotoProcessor:
                 result_image_standard_png,
                 result_image_hd_png,
                 gr.update(value=result_image_layout, visible=result_image_layout_visible),
+                gr.update(value=result_image_template, visible=result_image_template_visible),
             )
         else:
             # 如果output_image_path_dict不为None，
@@ -362,13 +372,15 @@ class IDPhotoProcessor:
                 (
                     output_image_path_dict["hd"]["path"]
                     if output_image_path_dict["hd"]["processed"]
-                    else result_image_hd
+                    else result_image_hd    
                 ),
                 result_image_standard_png,
                 result_image_hd_png,
                 gr.update(value=result_image_layout, visible=result_image_layout_visible),
+                gr.update(value=result_image_template, visible=result_image_template_visible),
             )
 
+    # 渲染背景
     def _render_background(self, result_image_standard, result_image_hd, idphoto_json):
         """渲染背景"""
         render_modes = {0: "pure_color", 1: "updown_gradient", 2: "center_gradient"}
@@ -387,21 +399,15 @@ class IDPhotoProcessor:
 
         return result_image_standard, result_image_hd
 
+    # 生成排版照片
     def _generate_image_layout(
         self,
         idphoto_json,
         result_image_standard,
         language,
-        watermark_option,
-        watermark_text,
-        watermark_text_size,
-        watermark_text_opacity,
-        watermark_text_angle,
-        watermark_text_space,
-        watermark_text_color,
     ):
         """生成排版照片"""
-        # 如果选择了自定义尺寸，则不生成排版照片
+        # 如果选择了只换底，则不生成排版照片
         if idphoto_json["size_mode"] in LOCALES["size_mode"][language]["choices"][1]:
             return None, False
 
@@ -409,21 +415,9 @@ class IDPhotoProcessor:
             input_height=idphoto_json["size"][0],
             input_width=idphoto_json["size"][1],
         )
-
-        image = result_image_standard
-        if watermark_option == LOCALES["watermark_switch"][language]["choices"][1]:
-            image = add_watermark(
-                image=image,
-                text=watermark_text,
-                size=watermark_text_size,
-                opacity=watermark_text_opacity,
-                angle=watermark_text_angle,
-                space=watermark_text_space,
-                color=watermark_text_color,
-            )
-
+        
         result_image_layout = generate_layout_image(
-            image,
+            result_image_standard,
             typography_arr,
             typography_rotate,
             height=idphoto_json["size"][0],
@@ -432,6 +426,28 @@ class IDPhotoProcessor:
 
         return result_image_layout, True
 
+    def _generate_image_template(
+        self,
+        idphoto_json,
+        result_image_hd,
+        language,
+    ):
+        # 如果选择了只换底，则不生成模板照片
+        if idphoto_json["size_mode"] in LOCALES["size_mode"][language]["choices"][1]:
+            return None, False
+        
+        TEMPLATE_NAME_LIST = ["template_1", "template_2"]
+        """生成模板照片"""
+        result_image_template_list = []
+        for template_name in TEMPLATE_NAME_LIST:
+            result_image_template = generte_template_photo(
+                template_name=template_name,
+                input_image=result_image_hd,
+            )
+            result_image_template_list.append(result_image_template)
+        return result_image_template_list, True
+
+    # 添加水印
     def _add_watermark(
         self,
         result_image_standard,
@@ -472,7 +488,10 @@ class IDPhotoProcessor:
         )
         timestamp = int(time.time())
         output_paths = {
-            "standard": {"path": f"{base_path}/{timestamp}_standard", "processed": False},
+            "standard": {
+                "path": f"{base_path}/{timestamp}_standard",
+                "processed": False,
+            },
             "hd": {"path": f"{base_path}/{timestamp}_hd", "processed": False},
             "layout": {"path": f"{base_path}/{timestamp}_layout", "processed": False},
         }
@@ -500,9 +519,11 @@ class IDPhotoProcessor:
             )
             output_paths["standard"]["processed"] = True
             # 保存高清图像和排版图像
-            save_image_dpi_to_bytes(result_image_hd, output_paths["hd"]["path"], dpi=custom_dpi)
+            save_image_dpi_to_bytes(
+                result_image_hd, output_paths["hd"]["path"], dpi=custom_dpi
+            )
             output_paths["hd"]["processed"] = True
-            if result_image_layout:
+            if result_image_layout is not None:
                 save_image_dpi_to_bytes(
                     result_image_layout, output_paths["layout"]["path"], dpi=custom_dpi
                 )
@@ -519,7 +540,9 @@ class IDPhotoProcessor:
                     pass
                 else:
                     save_image_dpi_to_bytes(
-                        locals()[f"result_image_{key}"], output_paths[key]["path"], dpi=custom_dpi
+                        locals()[f"result_image_{key}"],
+                        output_paths[key]["path"],
+                        dpi=custom_dpi,
                     )
                     output_paths[key]["processed"] = True
 
@@ -548,7 +571,8 @@ class IDPhotoProcessor:
         result_image_hd,
         result_image_standard_png,
         result_image_hd_png,
-        result_layout_image,
+        result_layout_image_gr,
+        result_image_template_gr,
     ):
         """创建响应"""
         response = [
@@ -556,7 +580,8 @@ class IDPhotoProcessor:
             result_image_hd,
             result_image_standard_png,
             result_image_hd_png,
-            result_layout_image,
+            result_layout_image_gr,
+            result_image_template_gr,
             gr.update(visible=False),
         ]
 

demo/ui.py

@@ -349,27 +349,34 @@ def create_ui(
                     label=LOCALES["notification"][DEFAULT_LANG]["label"], visible=False
                 )
                 with gr.Row():
+                    # 标准照
                     img_output_standard = gr.Image(
                         label=LOCALES["standard_photo"][DEFAULT_LANG]["label"],
                         height=350,
                         format="jpeg",
                     )
+                    # 高清照
                     img_output_standard_hd = gr.Image(
                         label=LOCALES["hd_photo"][DEFAULT_LANG]["label"],
                         height=350,
                         format="jpeg",
                     )
-
+                # 排版照
                 img_output_layout = gr.Image(
                     label=LOCALES["layout_photo"][DEFAULT_LANG]["label"],
                     height=350,
                     format="jpeg",
                 )
-
-                file_download = gr.File(
-                    label=LOCALES["download"][DEFAULT_LANG]["label"], visible=False
-                )
-
+                # 模版照片
+                with gr.Accordion(
+                    LOCALES["template_photo"][DEFAULT_LANG]["label"], open=False
+                ) as template_image_accordion:      
+                    img_output_template = gr.Gallery(
+                        label=LOCALES["template_photo"][DEFAULT_LANG]["label"],
+                        height=350,
+                        format="jpeg",
+                    )
+                # 抠图图像
                 with gr.Accordion(
                     LOCALES["matting_image"][DEFAULT_LANG]["label"], open=False
                 ) as matting_image_accordion:
@@ -443,9 +450,6 @@ def create_ui(
                     img_output_layout: gr.update(
                         label=LOCALES["layout_photo"][language]["label"]
                     ),
-                    file_download: gr.update(
-                        label=LOCALES["download"][language]["label"]
-                    ),
                     head_measure_ratio_option: gr.update(
                         label=LOCALES["head_measure_ratio"][language]["label"]
                     ),
@@ -530,6 +534,12 @@ def create_ui(
                     custom_size_height_mm: gr.update(
                         label=LOCALES["custom_size_mm"][language]["height"]
                     ),
+                    img_output_template: gr.update(
+                        label=LOCALES["template_photo"][language]["label"]
+                    ),
+                    template_image_accordion: gr.update(
+                        label=LOCALES["template_photo"][language]["label"]
+                    ),
                 }
 
             def change_visibility(option, lang, locales_key, custom_component):
@@ -612,7 +622,6 @@ def create_ui(
                     img_output_standard_png,
                     img_output_standard_hd_png,
                     img_output_layout,
-                    file_download,
                     head_measure_ratio_option,
                     top_distance_option,
                     key_parameter_tab,
@@ -639,6 +648,8 @@ def create_ui(
                     custom_size_height_px,
                     custom_size_width_mm,
                     custom_size_height_mm,
+                    img_output_template,
+                    template_image_accordion,
                 ],
             )
 
@@ -721,6 +732,7 @@ def create_ui(
                     img_output_standard_png,
                     img_output_standard_hd_png,
                     img_output_layout,
+                    img_output_template,
                     notification,
                 ],
             )

hivision/plugin/template/assets/template_config.json

@@ -0,0 +1,24 @@
+{
+    "template_1": {
+        "width": 1080,
+        "height": 1400,
+        "anchor_points": {
+            "left_top": [358, 153],
+            "right_top": [1017, 353],
+            "left_bottom": [56, 1134],
+            "right_bottom": [747, 1332],
+            "rotation": -16.42
+        }
+    },
+    "template_2": {
+        "width": 1080,
+        "height": 1440,
+        "anchor_points": {
+            "left_top": [199, 199],
+            "right_top": [921, 216],
+            "left_bottom": [163, 1129],
+            "right_bottom": [876, 1153],
+            "rotation": -2.2
+        }
+    }
+}

hivision/plugin/template/template_calculator.py

@@ -0,0 +1,78 @@
+import cv2
+import numpy as np
+import json
+from rotation_adjust import rotate_bound
+import os
+
+base_path = os.path.dirname(os.path.abspath(__file__))
+template_config_path = os.path.join(base_path, 'assets', 'template_config.json')
+
+def generte_template_photo(template_name: str, input_image: np.ndarray) -> np.ndarray:
+    """
+    生成模板照片
+    :param template_name: 模板名称
+    :param input_image: 输入图像
+    :return: 模板照片
+    """
+    # 读取模板配置json
+    with open(template_config_path, 'r') as f:
+        template_config_dict = json.load(f)
+    # 获取对应该模板的配置
+    template_config = template_config_dict[template_name]
+    
+    template_width = template_config['width']
+    template_height = template_config['height']
+
+    anchor_points = template_config['anchor_points']
+    rotation = anchor_points['rotation']
+    left_top = anchor_points['left_top']
+    right_top = anchor_points['right_top']
+    left_bottom = anchor_points['left_bottom']
+    right_bottom = anchor_points['right_bottom']
+
+    if rotation < 0:
+        height = right_bottom[1] - left_top[1]
+        width = right_top[0] - left_bottom[0]
+    else:
+        height = left_top[1] - right_bottom[1]
+        width = left_bottom[0] - right_top[0]
+
+    # 读取模板图像
+    template_image_path = os.path.join(base_path, 'assets', f'{template_name}.png')
+    template_image = cv2.imread(template_image_path, cv2.IMREAD_UNCHANGED)
+
+    # 无损旋转
+    rotated_image = rotate_bound(input_image, -1 * rotation)[0]
+    rotated_image_height, rotated_image_width, _ = rotated_image.shape
+
+    # 计算缩放比例
+    scale_x = width / rotated_image_width
+    scale_y = height / rotated_image_height
+    scale = max(scale_x, scale_y)
+
+    resized_image = cv2.resize(rotated_image, None, fx=scale, fy=scale)
+    resized_height, resized_width, _ = resized_image.shape
+
+    # 创建一个与template_image大小相同的背景，使用白色填充
+    result = np.full((template_height, template_width, 3), 255, dtype=np.uint8)
+
+    # 计算粘贴位置
+    paste_x = left_bottom[0]
+    paste_y = left_top[1]
+
+    # 确保不会超出边界
+    paste_height = min(resized_height, template_height - paste_y)
+    paste_width = min(resized_width, template_width - paste_x)
+
+    # 将旋转后的图像粘贴到结果图像上
+    result[paste_y:paste_y+paste_height, paste_x:paste_x+paste_width] = resized_image[:paste_height, :paste_width]
+    
+    template_image = cv2.cvtColor(template_image, cv2.COLOR_BGRA2RGBA)
+
+    # 将template_image叠加到结果图像上
+    if template_image.shape[2] == 4:  # 确保template_image有alpha通道
+        alpha = template_image[:, :, 3] / 255.0
+        for c in range(0, 3):
+            result[:, :, c] = result[:, :, c] * (1 - alpha) + template_image[:, :, c] * alpha
+
+    return result