hyzhou404 commited on
Commit
e050c1a
·
1 Parent(s): c9063a4
Files changed (1) hide show
  1. web_server.py +5 -5
web_server.py CHANGED
@@ -336,16 +336,16 @@ class EnvHandler:
336
  if not self._scene_done:
337
  return EnvExecuteResult(cur_scene_done=False, done=False)
338
 
339
- with open(os.path.join(self.cur_otuput, 'data.pkl'), 'wb') as wf:
340
  pickle.dump([self._save_data], wf)
341
 
342
- ground_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.cur_otuput, 'ground.ply')).points)
343
- scene_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.cur_otuput, 'scene.ply')).points)
344
  results = hugsim_evaluate([self._save_data], ground_xyz, scene_xyz)
345
- with open(os.path.join(self.cur_otuput, 'eval.json'), 'w') as f:
346
  json.dump(results, f)
347
  self._score_list.append(results.copy())
348
- to_video(self._observations_save, os.path.join(self.cur_otuput, 'video.mp4'))
349
 
350
  self._log(f"Scene {self.cur_scene_index} completed. Evaluation results saved.")
351
 
 
336
  if not self._scene_done:
337
  return EnvExecuteResult(cur_scene_done=False, done=False)
338
 
339
+ with open(os.path.join(self.cur_output, 'data.pkl'), 'wb') as wf:
340
  pickle.dump([self._save_data], wf)
341
 
342
+ ground_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.cur_output, 'ground.ply')).points)
343
+ scene_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.cur_output, 'scene.ply')).points)
344
  results = hugsim_evaluate([self._save_data], ground_xyz, scene_xyz)
345
+ with open(os.path.join(self.cur_output, 'eval.json'), 'w') as f:
346
  json.dump(results, f)
347
  self._score_list.append(results.copy())
348
+ to_video(self._observations_save, os.path.join(self.cur_output, 'video.mp4'))
349
 
350
  self._log(f"Scene {self.cur_scene_index} completed. Evaluation results saved.")
351