Spaces:
Paused
Paused
fix typo
Browse files- 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.
|
| 340 |
pickle.dump([self._save_data], wf)
|
| 341 |
|
| 342 |
-
ground_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.
|
| 343 |
-
scene_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.
|
| 344 |
results = hugsim_evaluate([self._save_data], ground_xyz, scene_xyz)
|
| 345 |
-
with open(os.path.join(self.
|
| 346 |
json.dump(results, f)
|
| 347 |
self._score_list.append(results.copy())
|
| 348 |
-
to_video(self._observations_save, os.path.join(self.
|
| 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 |
|