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

plot

Browse Source
This commit is contained in:
luni
2023-01-30 20:35:51 +00:00
parent 53b630c787
commit a82e1e7d9b
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
curvature_analysis.py
+9 -7
View File
@@ -163,16 +163,16 @@ def visualize_curvature_labels(folder, results_folder, frame):
fig.add_trace(go.Mesh3d( fig.add_trace(go.Mesh3d(
# 8 vertices of a cube # 8 vertices of a cube
x=verts[:,0], x=-pred[:,0],
y=verts[:,1], y=pred[:,2],
z=verts[:,2], z=pred[:,1],
# i, j and k give the vertices of triangles # i, j and k give the vertices of triangles
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=np.ones((verts.shape[0],3))*np.array([255,200,200])[None,:]
), ),
row=1, col=1) row=1, col=2)
fig.add_trace(go.Mesh3d( fig.add_trace(go.Mesh3d(
# 8 vertices of a cube # 8 vertices of a cube
@@ -185,7 +185,7 @@ def visualize_curvature_labels(folder, results_folder, frame):
k = faces[:,2], k = faces[:,2],
vertexcolor=colors_binned_all[0] vertexcolor=colors_binned_all[0]
), ),
row=1, col=2) row=1, col=1)
fig.add_trace(go.Mesh3d( fig.add_trace(go.Mesh3d(
# 8 vertices of a cube # 8 vertices of a cube
@@ -249,7 +249,7 @@ def init_wandb():
run.log_artifact(artifact) run.log_artifact(artifact)
if __name__ == '__main__': if __name__ == '__main__':
init_wandb() # init_wandb()
is_skirt = False is_skirt = False
skirt_modifier = '_skirt' if is_skirt else '' skirt_modifier = '_skirt' if is_skirt else ''
@@ -275,6 +275,8 @@ if __name__ == '__main__':
i = 0 i = 0
while True: while True:
frame = np.random.choice(frames) frame = np.random.choice(frames)
# frame = '03265_193'
frame = '03102_49'
fig, dice = visualize_curvature_labels(folder, results_folder, frame) fig, dice = visualize_curvature_labels(folder, results_folder, frame)
wandb.log({f'{frame}-dice:{dice}': fig}, step=i) wandb.log({f'{frame}-dice:{dice}': fig}, step=i)
i += 1 i += 1