From d231e1f5352358c7c7bd1a4cf6033a3d3cdf1781 Mon Sep 17 00:00:00 2001
From: las1g21 <las1g21@soton.ac.uk>
Date: Tue, 10 Dec 2024 22:41:10 +0000
Subject: [PATCH] Add .mtl to folder

---
 scripts/scene_completion.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/scripts/scene_completion.py b/scripts/scene_completion.py
index fdb3fcc..787832a 100644
--- a/scripts/scene_completion.py
+++ b/scripts/scene_completion.py
@@ -114,6 +114,11 @@ 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"
+        local_file_path = "edgenet-360/Data/Input/scene_completed_prediction.mtl"  
+        with SCPClient(client.get_transport()) as scp:
+            scp.get(remote_file_path, local_file_path)  # Download file
+
         print("OUTPUT DOWNLOADED")
         # Close the SSH connection 
         client.close() 
@@ -128,7 +133,7 @@ def get_completed_scene(shifted_disparity_path, shifted_t_path):
 
 if __name__ == "__main__":
     if len(sys.argv) != 3:
-        print("Usage: python shifter.py <shifted_disparity_path> <shifted_t_path>")
+        print("Usage: python scene_completion.py <shifted_disparity_path> <shifted_t_path>")
         sys.exit(1)
 
     shifted_disparity_path = sys.argv[1]
-- 
GitLab