File size: 877 Bytes
ddadf19 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
# coding: utf-8
__author__ = 'cleardusk'
import sys
sys.path.append('..')
import cv2
import numpy as np
from tddfa.Sim3DR import rasterize
from tddfa.utils.functions import plot_image
from .tddfa_util import _to_ctype
def depth(img, ver_lst, tri, show_flag=False, wfp=None, with_bg_flag=True):
if with_bg_flag:
overlap = img.copy()
else:
overlap = np.zeros_like(img)
for ver_ in ver_lst:
ver = _to_ctype(ver_.T) # transpose
z = ver[:, 2]
z_min, z_max = min(z), max(z)
z = (z - z_min) / (z_max - z_min)
# expand
z = np.repeat(z[:, np.newaxis], 3, axis=1)
overlap = rasterize(ver, tri, z, bg=overlap)
if wfp is not None:
cv2.imwrite(wfp, overlap)
# print(f'Save visualization result to {wfp}')
if show_flag:
plot_image(overlap)
return overlap
|