feat: update output code

web_server.py

@@ -28,6 +28,7 @@ IN_HUGGINGFACE_SPACE = os.getenv('IN_HUGGINGFACE_SPACE', 'false') == 'true'
 STOP_SPACE_TIMEOUT = int(os.getenv('STOP_SPACE_TIMEOUT', '7200'))
 HF_TOKEN = os.getenv('HF_TOKEN', None)
 SPACE_PARAMS = json.loads(os.getenv('PARAMS', '{}'))
+OUTPUT_DIR = "/app/app_datas/output"
 
 print("IN_HUGGINGFACE_SPACE:", IN_HUGGINGFACE_SPACE)
 print("STOP_SPACE_TIMEOUT:", STOP_SPACE_TIMEOUT)
@@ -104,6 +105,15 @@ def auto_stop():
     server_space_id = SPACE_PARAMS["server_space_id"]
     client_space_id = SPACE_PARAMS["client_space_id"]
     api = HfApi(token=HF_TOKEN)
+
+    if GlobalState.done:
+        api.upload_folder(
+            repo_id=SPACE_PARAMS["competition_id"],
+            folder_path=OUTPUT_DIR,
+            repo_type="dataset",
+            path_in_repo=f"eval_results/{SPACE_PARAMS['submission_id']}",
+        )
+
     api.delete_repo(
         repo_id=server_space_id,
         repo_type="space"
@@ -224,10 +234,10 @@ class EnvHandler:
         with open(os.path.join(self.output, 'data.pkl'), 'wb') as wf:
             pickle.dump([self._save_data], wf)
         
-        ground_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(output, 'ground.ply')).points)
-        scene_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(output, 'scene.ply')).points)
+        ground_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.output, 'ground.ply')).points)
+        scene_xyz = np.asarray(o3d.io.read_point_cloud(os.path.join(self.output, 'scene.ply')).points)
         results = hugsim_evaluate([self._save_data], ground_xyz, scene_xyz)
-        with open(os.path.join(output, 'eval.json'), 'w') as f:
+        with open(os.path.join(self.output, 'eval.json'), 'w') as f:
             json.dump(results, f)
         
         self._log("Evaluation results saved.")
@@ -335,18 +345,16 @@ if __name__ == "__main__":
         {"camera": camera_config},
         {"kinematic": kinematic_config}
     )
-    cfg.base.output_dir = cfg.base.output_dir + ad
 
     model_path = os.path.join(cfg.base.model_base, cfg.scenario.scene_name)
     model_config = OmegaConf.load(os.path.join(model_path, 'cfg.yaml'))
     model_config.update({"model_path": "/app/app_datas/PAMI2024/release/ss/scenes/nuscenes/scene-0383"})
     cfg.update(model_config)
     
-    output = os.path.join(cfg.base.output_dir, cfg.scenario.scene_name+"_"+cfg.scenario.mode)
-    os.makedirs(output, exist_ok=True)
+    os.makedirs(OUTPUT_DIR, exist_ok=True)
     
-    print("Output directory:", output)
-    env_handler = EnvHandler(cfg, output)
+    print("Output directory:", OUTPUT_DIR)
+    env_handler = EnvHandler(cfg, OUTPUT_DIR)
     print("Environment handler initialized.")
     web_server = WebServer(env_handler, auth_token=os.getenv('HUGSIM_AUTH_TOKEN'))
     print("Web server initialized.")