File size: 1,394 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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
# coding: utf-8
__author__ = 'cleardusk'
import sys
sys.path.append('..')
import cv2
import numpy as np
import os.path as osp
from Sim3DR import rasterize
from utils.functions import plot_image
from utils.io import _load, _dump
from utils.tddfa_util import _to_ctype
make_abs_path = lambda fn: osp.join(osp.dirname(osp.realpath(__file__)), fn)
def calc_ncc_code():
from bfm import bfm
# formula: ncc_d = ( u_d - min(u_d) ) / ( max(u_d) - min(u_d) ), d = {r, g, b}
u = bfm.u
u = u.reshape(3, -1, order='F')
for i in range(3):
u[i] = (u[i] - u[i].min()) / (u[i].max() - u[i].min())
_dump('../configs/ncc_code.npy', u)
def pncc(img, ver_lst, tri, show_flag=False, wfp=None, with_bg_flag=True):
ncc_code = _load(make_abs_path('../configs/ncc_code.npy'))
if with_bg_flag:
overlap = img.copy()
else:
overlap = np.zeros_like(img)
# rendering pncc
for ver_ in ver_lst:
ver = _to_ctype(ver_.T) # transpose
overlap = rasterize(ver, tri, ncc_code.T, bg=overlap) # m x 3
if wfp is not None:
cv2.imwrite(wfp, overlap)
print(f'Save visualization result to {wfp}')
if show_flag:
plot_image(overlap)
return overlap
def main():
# `configs/ncc_code.npy` is generated by `calc_nnc_code` function
# calc_ncc_code()
pass
if __name__ == '__main__':
main()
|