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'
|