Merge branch 'fix-mbdnet-file-naming' into 'master'

file names fixed

See merge request !20

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} ')

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: