Add MBDNet tab to GUI and update file paths in scene completion script

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,12 +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 = "edgenet-360/Data/Input/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