yejunliang23 commited on
Commit
9a346a6
·
verified ·
1 Parent(s): 7e9dcdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -60
app.py CHANGED
@@ -284,69 +284,14 @@ def reset_state(task_history):
284
  task_history.clear()
285
  return []
286
 
287
- def make_pointcloud_figure(verts,rotate=False):
288
  fig = go.Figure(go.Scatter3d(
289
  x=[0.005*n for n in range(100)], y=[0.005*n for n in range(100)], z=[0.005*n for n in range(100)],
290
  mode='markers', marker=dict(size=8)
291
  ))
292
  return fig
293
 
294
- def make_pointcloud_figure_new1(verts, rotate=False):
295
- """
296
- Simple 3D scatter of point cloud that always shows points.
297
-
298
- Parameters:
299
- -------------
300
- verts : (N, 3) numpy array or torch tensor
301
- Point cloud coordinates.
302
- rotate : bool
303
- If True, reflect X axis.
304
-
305
- Returns:
306
- --------
307
- fig : plotly.graph_objects.Figure
308
- """
309
- # Convert to numpy if torch tensor
310
- try:
311
- import torch
312
- if isinstance(verts, torch.Tensor):
313
- verts = verts.cpu().numpy()
314
- except ImportError:
315
- pass
316
-
317
- if rotate:
318
- verts = verts.copy()
319
- verts[:, 0] *= -1.0
320
-
321
- # Build scatter trace with a fixed color and larger size
322
- scatter = go.Scatter3d(
323
- x=verts[:, 0],
324
- y=verts[:, 1],
325
- z=verts[:, 2],
326
- mode='markers',
327
- marker=dict(
328
- size=5,
329
- color='blue',
330
- opacity=0.8
331
- )
332
- )
333
-
334
- # Default layout: auto ranging, hidden axes, minimal margins
335
- layout = go.Layout(
336
- scene=dict(
337
- xaxis=dict(visible=False),
338
- yaxis=dict(visible=False),
339
- zaxis=dict(visible=False),
340
- aspectmode='auto'
341
- ),
342
- margin=dict(l=0, r=0, b=0, t=0)
343
- )
344
-
345
- fig = go.Figure(data=[scatter], layout=layout)
346
- return fig
347
-
348
-
349
- def make_pointcloud_figure_old(verts,rotate=False):
350
  if rotate:
351
  verts = verts.copy()
352
  verts[:, 0] *= -1.0
@@ -364,10 +309,12 @@ def make_pointcloud_figure_old(verts,rotate=False):
364
  projection=dict(type="orthographic")
365
  )
366
 
 
 
367
  scatter = go.Scatter3d(
368
- x=verts[:, 0],
369
- y=verts[:, 1],
370
- z=verts[:, 2],
371
  mode='markers',
372
  marker=dict(
373
  size=2,
 
284
  task_history.clear()
285
  return []
286
 
287
+ def make_pointcloud_figure_success(verts,rotate=False):
288
  fig = go.Figure(go.Scatter3d(
289
  x=[0.005*n for n in range(100)], y=[0.005*n for n in range(100)], z=[0.005*n for n in range(100)],
290
  mode='markers', marker=dict(size=8)
291
  ))
292
  return fig
293
 
294
+ def make_pointcloud_figure(verts,rotate=False):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
295
  if rotate:
296
  verts = verts.copy()
297
  verts[:, 0] *= -1.0
 
309
  projection=dict(type="orthographic")
310
  )
311
 
312
+ print(len(verts[:, 0].tolist()))
313
+ print(verts[:, 0].tolist())
314
  scatter = go.Scatter3d(
315
+ x=verts[:, 0].tolist(),
316
+ y=verts[:, 1].tolist(),
317
+ z=verts[:, 2].tolist(),
318
  mode='markers',
319
  marker=dict(
320
  size=2,