Fix PNCC

cleardusk · Dec 27, 2018 · d4f758e · d4f758e
1 parent c320de1
commit d4f758e
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/samples/demo_pncc_paf.jpg b/samples/demo_pncc_paf.jpg
diff --git a/train.configs/pncc_code.npy b/train.configs/pncc_code.npy
diff --git a/utils/params.py b/utils/params.py
@@ -38,3 +38,6 @@ def make_abs_path(d):
 u_filter = paf.get('mu_filter')
 w_filter = paf.get('w_filter')
 w_exp_filter = paf.get('w_exp_filter')
+
+# pncc code (mean shape)
+pncc_code = _load(osp.join(d, 'pncc_code.npy'))
diff --git a/utils/render.py b/utils/render.py
@@ -15,6 +15,7 @@
 
 import numpy as np
 from .cython import mesh_core_cython
+from .params import pncc_code
 
 
 def is_point_in_tri(point, tri_points):
@@ -189,6 +190,21 @@ def ncc(vertices):
 
 
 def cpncc(img, vertices_lst, tri):
+    """cython version for PNCC render: original paper"""
+    h, w = img.shape[:2]
+    c = 3
+
+    pnccs_img = np.zeros((h, w, c))
+    for i in range(len(vertices_lst)):
+        vertices = vertices_lst[i]
+        pncc_img = crender_colors(vertices, tri, pncc_code, h, w, c)
+        pnccs_img[pncc_img > 0] = pncc_img[pncc_img > 0]
+
+    pnccs_img = pnccs_img.squeeze() * 255
+    return pnccs_img
+
+
+def cpncc_v2(img, vertices_lst, tri):
     """cython version for PNCC render"""
     h, w = img.shape[:2]
     c = 3