From 00388354b45e493e48d418ac74d1dcde8cc0adaa Mon Sep 17 00:00:00 2001
From: las1g21 <las1g21@soton.ac.uk>
Date: Mon, 20 Jan 2025 13:56:04 +0000
Subject: [PATCH] file names fixed
---
scripts/debug_tool/tabs/mbdnet_tab.py | 4 ++--
scripts/simple_tab.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/scripts/debug_tool/tabs/mbdnet_tab.py b/scripts/debug_tool/tabs/mbdnet_tab.py
index 165d59d..015ed34 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 f4b69bd..cb9eda5 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:
--
GitLab