Spaces:
Build error
Build error
Ren Jiawei
commited on
Commit
·
d015e2a
1
Parent(s):
ce7f652
update
Browse files
app.py
CHANGED
|
@@ -84,7 +84,8 @@ def recognize_pcd(model, pcd):
|
|
| 84 |
|
| 85 |
def run(seed, corruption_idx, severity):
|
| 86 |
data, label = load_dataset(corruption_idx, severity)
|
| 87 |
-
|
|
|
|
| 88 |
pcd, cls = data[sample_indx], label[sample_indx]
|
| 89 |
pyplot_draw_point_cloud(pcd, CLASS_NAME[cls[0]])
|
| 90 |
output = 'visualization.png'
|
|
|
|
| 84 |
|
| 85 |
def run(seed, corruption_idx, severity):
|
| 86 |
data, label = load_dataset(corruption_idx, severity)
|
| 87 |
+
random.seed(seed)
|
| 88 |
+
sample_indx = random.randint(0, data.shape[0])
|
| 89 |
pcd, cls = data[sample_indx], label[sample_indx]
|
| 90 |
pyplot_draw_point_cloud(pcd, CLASS_NAME[cls[0]])
|
| 91 |
output = 'visualization.png'
|