Merge branch 'creating-mbdnet-tab' into 'master'

Creating mbdnet tab

See merge request !14

scripts/debug_tool/GUI_debug.py

@@ -9,6 +9,7 @@ from tabs.depth_tab import DepthTab
 from tabs.material_tab import MaterialTab
 from tabs.edge_net_tab import EdgeNetTab
 from utils.config_reader import ConfigReader
+from tabs.mbdnet_tab import MBDNetTab
 
 class ModuleDebugGUI(QMainWindow):
     def __init__(self):
@@ -43,6 +44,7 @@ class ModuleDebugGUI(QMainWindow):
         self.tabs.addTab(DepthTab(self.config_reader), "MonoDepth Estimation")
         self.tabs.addTab(MaterialTab(self.config_reader), "Material Recognition")
         self.tabs.addTab(EdgeNetTab(self.config_reader), "EdgeNet Execution")
+        self.tabs.addTab(MBDNetTab(self.config_reader), "MBDNet Execution")
 
 def main():
     app = QApplication(sys.argv)

scripts/scene_completion.py

@@ -4,11 +4,15 @@ from scp import SCPClient
 import os
 import subprocess
 import time
+import sys
 
 load_dotenv()
 hostname = os.getenv("HOSTNAME")
-username = os.getenv("USERNAME")
+#username = os.getenv("USERNAME")
 password = os.getenv("PASSWORD")
+# hostname = "soton.ac.uk"
+username = "kproject"
+# password = "i7-13700H"
 
 def send_files(shifted_disparity_path, shifted_t_path):
     # sends two files needed to iridis
@@ -110,7 +114,15 @@ def get_completed_scene(shifted_disparity_path, shifted_t_path):
             print(output)
 
         remote_file_path = "/mainfs/ECShome/kproject/mona/MDBNet360_GDP/output/scene_completed_prediction.obj"
-        local_file_path = "edgenet-360/Data/Input/scene_completed_prediction.obj"  
+        current_dir = os.getcwd()
+        parent_dir = os.path.dirname(current_dir)
+        local_file_folder = os.path.join(parent_dir, "edgenet-360/Output")
+        local_file_path = os.path.join(local_file_folder,"scene_completed_prediction.obj")  
+        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 = 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
 
@@ -128,11 +140,11 @@ 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]
     shifted_t_path = sys.argv[2]
 
-    saved_path = get_completed_scene(shifted_disparity_path, shifted_t_path)
+    saved_path = get_completed_scene(shifted_disparity_path, shifted_t_path)[1]
     print(f"Shifted image saved as {saved_path}")
\ No newline at end of file