Ren Jiawei commited on
Commit
d015e2a
1 Parent(s): ce7f652
Files changed (1) hide show
  1. app.py +2 -1
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
- sample_indx = int(seed)
 
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'