hofee/nbv_rec_blender_render
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

upd

Browse Source
This commit is contained in:
hofee 2024-11-05 22:01:41 +08:00
parent 0ba46d4402
commit 4f523b20d5
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
data_generator.py
View File

@ -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: