luni/cvpr_2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add uv normal example

Browse Source
This commit is contained in:
luni
2023-07-04 18:23:36 +02:00
parent 8dc6e0c71a
commit 4fca0f1c4f
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/generate_figures.py
+10 -2
View File
@@ -12,6 +12,7 @@ from data.cloth3d_io import readPC2Frame
from data.utils import Frame
from custom_utils.custom_utils import _maskFaces
import plotly.graph_objects as go
import plotly.express as ex
from plotly.subplots import make_subplots
@@ -72,7 +73,7 @@ def plot_normals_example():
# mesh_unposed = readPC2Frame(os.path.join(folder_test, 'mesh_unposed_no_shape', f'{frame.id}_{frame.garment}.pc16'), frame.frame_nr, True)
# mesh_unposed_shaped = readPC2Frame(os.path.join('temp', f'{frame.id}_{frame.garment}_unposed_shaped.pc16'), frame.frame_nr, True)[mesh_mask]
mesh_normals = readPC2Frame(os.path.join(folder_test, 'mesh_normals', f'{frame.id}_{frame.garment}.pc16'), frame.frame_nr, True)
# uv_normals = np.load(os.path.join(folder_test, 'uv_normals', f'{frame.id}_{frame.frame_nr}_{frame.garment}.npz'))
uv_normals = np.load(os.path.join(folder_test, 'uv_normals', f'{frame.id}_{frame.frame_nr}_{frame.garment}.npz'))['uv_normals']
F = np.load('smpl/smpl_faces.npy')
_, faces = _maskFaces(np.zeros((14475, 3)), F, mesh_mask)
@@ -105,7 +106,14 @@ def plot_normals_example():
zaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),)
fig.update_layout(scene = helper)
fig.show()
# fig.show()
uv_normals = (uv_normals + 1.4) / 2.8
uv_normals = np.concatenate((uv_normals[:256], uv_normals[256:]), axis=1)
# flip uv_normals along y axis
uv_normals = uv_normals[:,::-1,:]
ex.imshow(uv_normals).show()