Skip to content
Snippets Groups Projects
Commit 65f071d0 authored by mhby1g21's avatar mhby1g21
Browse files

fixed edgenet and post processes and config reader init in GUI.py

parent f4c29946
No related branches found
No related tags found
1 merge request!74.3.1 - Redesign GUI and remove .bat dependencies and GDP 4.3.3 - Delete Temporary Files On Main Pipeline Run on new GUI.py
...@@ -26,7 +26,7 @@ class VRSceneCreatorGUI(QMainWindow): ...@@ -26,7 +26,7 @@ class VRSceneCreatorGUI(QMainWindow):
self.DEBUG_TOOL_DIR = DEBUG_TOOL_DIR self.DEBUG_TOOL_DIR = DEBUG_TOOL_DIR
# Read configuration # Read configuration
self.config_reader = ConfigReader(self.SCRIPT_DIR, self.ROOT_DIR) self.config_reader = ConfigReader(self.DEBUG_TOOL_DIR, self.ROOT_DIR)
# Setup UI # Setup UI
self.setup_ui() self.setup_ui()
......
...@@ -102,11 +102,24 @@ class PipelineWorker(QThread): ...@@ -102,11 +102,24 @@ class PipelineWorker(QThread):
print("Completed material_recognition") print("Completed material_recognition")
def run_edge_net(self): def run_edge_net(self):
print("Starting edge_net") # Debug print print("Starting edge_net")
self.progress.emit("Running EdgeNet...") self.progress.emit("Running EdgeNet enhance360.py and infer360.py...")
try:
self.tab.edge_net.include_top = self.tab.include_top_check.isChecked() self.tab.edge_net.include_top = self.tab.include_top_check.isChecked()
self.tab.edge_net.run_all_steps() self.tab.edge_net._run_edge_net_process()
print("Completed edge_net") # Debug print print("Completed edge_net")
except Exception as e:
print(f"EdgeNet failed: {str(e)}")
raise
def run_post_processing(self):
self.progress.emit("Running post-processing...")
self.tab.edge_net._run_mesh_split_process()
self.progress.emit("Completed mesh split (replace.py)")
self.tab.edge_net._run_blender_flip_process()
self.progress.emit("Completed blender flip (blenderFlip.py)")
self.progress.emit("Post-processing completed!")
def run_pipeline(self): def run_pipeline(self):
self.clean_temp_files() self.clean_temp_files()
...@@ -115,6 +128,8 @@ class PipelineWorker(QThread): ...@@ -115,6 +128,8 @@ class PipelineWorker(QThread):
self.run_depth_estimation() self.run_depth_estimation()
self.run_material_recognition() self.run_material_recognition()
self.run_edge_net() self.run_edge_net()
self.run_post_processing()
self.progress.emit("Pipeline completed!")
class SimpleTab(QWidget): class SimpleTab(QWidget):
def __init__(self, config_reader): def __init__(self, config_reader):
...@@ -201,10 +216,12 @@ class SimpleTab(QWidget): ...@@ -201,10 +216,12 @@ class SimpleTab(QWidget):
layout.addWidget(preview_group) layout.addWidget(preview_group)
def handle_file_select(self): def handle_file_select(self):
file_path = select_file( file_path = select_file(
self, self,
"Select Input Image", "Select Input Image",
"Images (*.png *.jpg *.jpeg)" "Images (*.png *.jpg *.jpeg)",
initial_dir=self.config_reader.directories['edgeNetDir'] + '/Data'
) )
if file_path: if file_path:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment