alpha到trimap（python）

哇噻YL.

已于 2024-01-11 15:02:44 修改

阅读量47

点赞数

文章标签： python

于 2021-03-10 15:29:02 首次发布

本文链接：https://blog.csdn.net/weixin_44522051/article/details/114636873

版权

alpha到trimap（python）

import os
import numpy as np
import cv2

def random_dilate(alpha, low=2, high=3, mode='constant'):
    iterations = np.random.randint(1, 2)
    erode_ksize = np.random.randint(low=low, high=high)
    dilate_ksize = np.random.randint(low=low, high=high)
    erode_kernel = cv2.getStructuringElement(
        cv2.MORPH_ELLIPSE, (erode_ksize, erode_ksize))
    dilate_kernel = cv2.getStructuringElement(
        cv2.MORPH_ELLIPSE, (dilate_ksize, dilate_ksize))
    alpha_eroded = cv2.erode(alpha, erode_kernel, iterations=iterations)
    alpha_dilated = cv2.dilate(alpha, dilate_kernel, iterations=iterations)
    if mode == 'constant':
        alpha_noise = 128 * np.ones_like(alpha)
        alpha_noise[alpha_eroded >= 225] = 255
        alpha_noise[alpha_dilated <= 128] = 0
    else:
        value = np.random.randint(low=100, high=255)
        alpha_noise = value * ((alpha_dilated - alpha_eroded) / 255.)
        alpha_noise += alpha_eroded
    return alpha_noise