## 如何做一个好的边缘过渡
import torch
import torch.nn as nn
import numpy as np
import cv2
import matplotlib.pyplot as plt
depth = cv2.imread('.png')
seg = cv2.imread('.png', cv2.IMREAD_UNCHANGED)
alpha = seg[:, :, 3:].repeat(3, 2).copy()
zero_length = 10
alpha[0:zero_length] = 0
alpha[-zero_length:] = 0
alpha[:, 0:zero_length] = 0
alpha[:, -zero_length:] = 0
alpha_inv = 255-alpha
alpha_blur = alpha_inv.copy()
for _ in range(25):
# alpha_blur = cv2.GaussianBlur(alpha_inv, (11,11), 0)
alpha_blur = cv2.blur(alpha_inv, (51, 51))
alpha_inv[np.where(alpha>0)] = alpha_blur[np.where(alpha>0)]
alpha_new = (255-alpha_inv)/255.0
depth = (depth*alpha_new).astype(np.uint8)
cv2.imwrite('test.png', depth)
如何做一个好的边缘过渡
最新推荐文章于 2024-09-07 20:08:36 发布