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