生成三分图trimap及前景背景融合

参考1参考2

一 生成三分图trimap

import random
import numpy as np 
import cv2

def gen_trimap(alpha):
    k_size = random.choice(range(1, 5)) # 随机选择核大小
    iterations = np.random.randint(1, 20)
    ###cv2.getStructuringElement( )
    ###这个函数的第一个参数表示内核的形状，有三种形状可以选择。矩形：MORPH_RECT;交叉形：MORPH_CROSS;椭圆形：MORPH_ELLIPSE
    ###第二和第三个参数分别是内核的尺寸以及锚点的位置。一般在调用erode以及dilate函数之前，先定义一个Mat类型的变量来获得
    ###getStructuringElement函数的返回值: 对于锚点的位置，有默认值Point（-1,-1），表示锚点位于中心点。element形状唯一依赖锚点位置，其他情况下，锚点只是影响了形态学运算结果的偏移。
    kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (k_size, k_size))
    
    dilated = cv2.dilate(alpha, kernel, iterations)
    eroded = cv2.erode(alpha, kernel, iterations)
    trimap = np.zeros(alpha.shape)
    trimap.fill(128)
    trimap[eroded >= 255] = 255
    trimap[dilated <= 0] = 0
    
    return trimap

if __name__ == "__main__":
    alpha = xxxx
    assert (np.unique(alpha_np) == [0, 255]).any()
    alpha = alpha * 255
    gen_trimap(alpha)