diff --git a/scripts/scene_completion.py b/scripts/scene_completion.py index 6baef21f331197d3d88ea8cebfb0e564d8e4e9ff..e3514d42c20f249760a19f42177b7ee379b3f4a0 100644 --- a/scripts/scene_completion.py +++ b/scripts/scene_completion.py @@ -113,7 +113,7 @@ def get_completed_scene(shifted_disparity_path, shifted_t_path): output = stdout.read().decode() print(output) - remote_file_path = "/mainfs/ECShome/kproject/mona/MDBNet360_GDP/output/scene_completed_prediction.obj" + remote_file_path = os.getenv("REMOTE_OUTPUT_PATH_OBJ") current_dir = os.getcwd() parent_dir = os.path.dirname(current_dir) local_file_folder = os.path.join(parent_dir, "edgenet-360/Output") @@ -121,7 +121,7 @@ def get_completed_scene(shifted_disparity_path, shifted_t_path): with SCPClient(client.get_transport()) as scp: scp.get(remote_file_path, local_file_path) # Download file - remote_file_path = "/mainfs/ECShome/kproject/mona/MDBNet360_GDP/output/scene_completed_prediction.mtl" + remote_file_path = os.getenv("REMOTE_OUTPUT_PATH_MTL") local_file_path = os.path.join(local_file_folder,"scene_completed_prediction.mtl") with SCPClient(client.get_transport()) as scp: scp.get(remote_file_path, local_file_path) # Download file