32 lines
859 B
Python
32 lines
859 B
Python
|
#!/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("")
|