diff --git a/scripts/debug_tool/tabs/mbdnet_tab.py b/scripts/debug_tool/tabs/mbdnet_tab.py index 165d59d067595a3ad70d4a746471c2c86bcf9219..015ed344f7ec5bde265363603a783e5a76086a4c 100644 --- a/scripts/debug_tool/tabs/mbdnet_tab.py +++ b/scripts/debug_tool/tabs/mbdnet_tab.py @@ -290,7 +290,7 @@ class MBDNetTab(QWidget): #self.clean_output_directory(self.output_dir, True) # copy rgb.png to shifted_t.png and enhanced_depth_e.png to shifted_disparity.png os.rename(os.path.join(self.input_dir, 'rgb.png'), os.path.join(self.input_dir, 'shifted_t.png')) - os.rename(os.path.join(self.input_dir, 'depth_e.png'), os.path.join(self.input_dir, 'shifted_disparity.png')) + os.rename(os.path.join(self.input_dir, 'depth_e.png'), os.path.join(self.input_dir, 'shifted-disparity.png')) # Run scene_completion.py os.chdir(self.scripts_dir) infer_cmd = self._build_mbdnet_command() @@ -309,7 +309,7 @@ class MBDNetTab(QWidget): def _build_mbdnet_command(self): """Build scene_completion.py command""" - shifted_disparity = os.path.join(self.input_dir, 'shifted_disparity.png') + shifted_disparity = os.path.join(self.input_dir, 'shifted-disparity.png') shifted_t = os.path.join(self.input_dir, 'shifted_t.png') command = (f'python scene_completion.py {shifted_disparity} {shifted_t} ') diff --git a/scripts/simple_tab.py b/scripts/simple_tab.py index f4b69bd9fb818e03122145e2531b060c78eeafdd..cb9eda509e37eb33e0749ddbd4a1e93ed5661b5b 100644 --- a/scripts/simple_tab.py +++ b/scripts/simple_tab.py @@ -205,8 +205,8 @@ class PipelineWorker(QThread): else: self.run_depth_estimation() #self.run_depth_to_disparity() - self.run_material_recognition() if self.edgenet_flag: + self.run_material_recognition() self.run_edge_net() self.run_post_processing() else: