From 4f523b20d59b58b9ef6caeb3226065fa1a9a6fec Mon Sep 17 00:00:00 2001
From: hofee <64160135+GitHofee@users.noreply.github.com>
Date: Tue, 5 Nov 2024 22:01:41 +0800
Subject: [PATCH] upd

---
 data_generator.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/data_generator.py b/data_generator.py
index 87a4518..4293148 100644
--- a/data_generator.py
+++ b/data_generator.py
@@ -18,6 +18,7 @@ class DataGenerator:
         self.random_config = config["runner"]["generate"]["random_config"]
         self.light_and_camera_config = config["runner"]["generate"]["light_and_camera_config"]
         self.obj_dir = config["runner"]["generate"]["object_dir"]
+        
         self.max_views = config["runner"]["generate"]["max_views"]
         self.min_views = config["runner"]["generate"]["min_views"]
         self.min_diag = config["runner"]["generate"]["min_diag"]
@@ -30,6 +31,9 @@ class DataGenerator:
         self.to_idx = config["runner"]["generate"]["to"]
         self.set_status_path = f"http://localhost:{self.port}/project/set_status"
         self.log_path = f"http://localhost:{self.port}/project/add_log"
+        
+        if not os.path.exists(self.output_dir):
+            os.makedirs(self.output_dir)
         self.obj_name_list = os.listdir(self.obj_dir)[self.from_idx: self.to_idx]
         self.target_obj = None
         self.stopped = False
@@ -243,11 +247,7 @@ class DataGenerator:
 
         msg = self.check_and_adjust_target()
         
-        if msg == "adjusted" and depth < 3:
-            bpy.context.view_layer.update()
-            bpy.context.scene.frame_set(0)
-            return self.simulate_scene(depth = depth + 1, diag=diag)
-        elif msg == "success":
+        if msg == "success":
             print("Scene generation completed.")
             result = self.start_render(diag=diag)
             if not result: