Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -284,69 +284,14 @@ def reset_state(task_history):
|
|
284 |
task_history.clear()
|
285 |
return []
|
286 |
|
287 |
-
def
|
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
|
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,
|