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

imporve plot

Browse Source
This commit is contained in:
luni
2023-01-30 11:18:59 +00:00
parent 798bf1367d
commit 11a8c7e63e
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
curvature_analysis.py
+10 -5
View File
@@ -133,6 +133,8 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
fig = make_subplots(rows=1, cols=len(colors_binned_all)+1, start_cell="top-left", specs = [ [{ 'type': 'scene' } for _ in range(len(colors_binned_all)+1)] ]) fig = make_subplots(rows=1, cols=len(colors_binned_all)+1, start_cell="top-left", specs = [ [{ 'type': 'scene' } for _ in range(len(colors_binned_all)+1)] ])
colors = np.ones((verts.shape[0],3))*np.array([255,200,200])[None,:]
fig.add_trace(go.Mesh3d( fig.add_trace(go.Mesh3d(
# 8 vertices of a cube # 8 vertices of a cube
x=verts[:,0], x=verts[:,0],
@@ -142,7 +144,8 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
i = faces[:,0], i = faces[:,0],
j = faces[:,1], j = faces[:,1],
k = faces[:,2], k = faces[:,2],
vertexcolor=np.ones((verts.shape[0],3))*200 vertexcolor=colors,
lighting={ 'ambient':0.9, 'diffuse':0.5, 'roughness':0.5 }
), ),
row=1, col=1) row=1, col=1)
@@ -156,7 +159,8 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
i = faces[:,0], i = faces[:,0],
j = faces[:,1], j = faces[:,1],
k = faces[:,2], k = faces[:,2],
vertexcolor=colors_binned vertexcolor=colors_binned,
# lighting={ 'ambient':0.8, 'diffuse':0.5, 'roughness':0.5 }
), ),
row=1, col=i+2) row=1, col=i+2)
@@ -168,9 +172,10 @@ def visualize_curvature_labels(folder, frame, bins, neighbourhoods):
scene4_aspectmode='manual', scene4_aspectratio=dict(x=1, y=1, z=1), scene4_aspectmode='manual', scene4_aspectratio=dict(x=1, y=1, z=1),
# scene5_aspectmode='manual', scene5_aspectratio=dict(x=1, y=1, z=1), # scene5_aspectmode='manual', scene5_aspectratio=dict(x=1, y=1, z=1),
) )
helper = dict(xaxis = dict(nticks=4, range=[-1,1],), helper = dict(xaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),
yaxis = dict(nticks=4, range=[-1,1],), yaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),
zaxis = dict(nticks=4, range=[-1,1],),) zaxis = dict(range=[-1,1], showbackground=False, showticklabels=False, showaxeslabels=False, showspikes=False, visible=False),)
fig.update_layout(scene = helper, fig.update_layout(scene = helper,
scene2 = helper, scene2 = helper,
scene3 = helper, scene3 = helper,