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
Branches
Tags
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):
self.DEBUG_TOOL_DIR = DEBUG_TOOL_DIR
# 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
self.setup_ui()
......
......@@ -102,11 +102,24 @@ class PipelineWorker(QThread):
print("Completed material_recognition")
def run_edge_net(self):
print("Starting edge_net") # Debug print
self.progress.emit("Running EdgeNet...")
print("Starting edge_net")
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.run_all_steps()
print("Completed edge_net") # Debug print
self.tab.edge_net._run_edge_net_process()
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):
self.clean_temp_files()
......@@ -115,6 +128,8 @@ class PipelineWorker(QThread):
self.run_depth_estimation()
self.run_material_recognition()
self.run_edge_net()
self.run_post_processing()
self.progress.emit("Pipeline completed!")
class SimpleTab(QWidget):
def __init__(self, config_reader):
......@@ -201,10 +216,12 @@ class SimpleTab(QWidget):
layout.addWidget(preview_group)
def handle_file_select(self):
file_path = select_file(
self,
"Select Input Image",
"Images (*.png *.jpg *.jpeg)"
"Images (*.png *.jpg *.jpeg)",
initial_dir=self.config_reader.directories['edgeNetDir'] + '/Data'
)
if file_path:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment