design batch 代码整理

app/service/design_batch/utils/conversion_image.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+"""
+@Project ：trinity_client
+@File    ：conversion_image.py
+@Author  ：周成融
+@Date    ：2023/8/21 10:40:29
+@detail  ：
+"""
+import numpy as np
+
+
+# def rgb_to_rgba(rgb_size, rgb_image, mask):
+#     alpha_channel = np.full(rgb_size, 255, dtype=np.uint8)
+#     # 创建四通道的结果图像
+#     rgba_image = np.dstack((rgb_image, alpha_channel))
+#     alpha_channel = np.where(mask > 0, 255, 0)
+#     # 更新RGBA图像的透明度通道
+#     rgba_image[:, :, 3] = alpha_channel
+#     return rgba_image
+
+def rgb_to_rgba(rgb_image, mask):
+    # 创建全透明的alpha通道
+    alpha_channel = np.where(mask > 0, 255, 0).astype(np.uint8)
+    # 合并RGB图像和alpha通道
+    rgba_image = np.dstack((rgb_image, alpha_channel))
+    return rgba_image
+
+
+if __name__ == '__main__':
+    image = open("")