Skip to content
Snippets Groups Projects
Commit 00388354 authored by las1g21's avatar las1g21
Browse files

file names fixed

parent 35fd82cd
No related branches found
No related tags found
1 merge request!20file names fixed
......@@ -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} ')
......
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment