init commit. change the environment to be more clutter. change the default desk height and the door width

Author: yuqiang-yang

count_obj.py

@@ -0,0 +1,41 @@
+def count_lines_per_object(obj_file_path):
+    object_lines = {}
+    current_object = None
+    line_count = 0
+
+    try:
+        with open(obj_file_path, 'r') as file:
+            lines = file.readlines()
+            for i, line in enumerate(lines):
+                if line.startswith('o '):
+                    if current_object is not None:
+                        object_lines[current_object] = line_count
+                    current_object = line.split(' ', 1)[1].strip()
+                    line_count = 0
+                line_count += 1
+
+            # 处理最后一个物体
+            if current_object is not None:
+                object_lines[current_object] = line_count
+
+    except FileNotFoundError:
+        print(f"文件 {obj_file_path} 未找到。")
+        return
+
+    return object_lines
+
+# 示例使用
+obj_file_path = '/ssd/yangyuqiang/infinigen/outputs/multi_dataset_no_plantandshlefobj/506fc8f/fine/scene.obj'
+result = count_lines_per_object(obj_file_path)
+
+# 计算总行数
+total_lines = sum(result.values())
+
+# 按照行数降序排序
+sorted_result = sorted(result.items(), key=lambda item: item[1], reverse=True)
+total_pro = 0.0
+for object_name, line_count in sorted_result:
+    # 计算单个物体占总物体大小的比例
+    proportion = (line_count / total_lines) * 100
+    total_pro += proportion
+    print(f"物体 {object_name} 所占行数: {line_count}，占比: {proportion:.5f}%, cumsum: {total_pro:.5f}% ")

infinigen/assets/objects/shelves/simple_desk.py

@@ -410,7 +410,7 @@ def create_asset(self, i=0, **params):
 class SimpleDeskFactory(SimpleDeskBaseFactory):
     def sample_params(self):
         params = dict()
-        params["Dimensions"] = (uniform(0.5, 0.75), uniform(0.8, 2), uniform(0.6, 0.8))
+        params["Dimensions"] = (uniform(0.5, 0.75), uniform(0.8, 2), uniform(0.8, 1.2))
         params["depth"] = params["Dimensions"][0]
         params["width"] = params["Dimensions"][1]
         params["height"] = params["Dimensions"][2]

infinigen/core/constraints/constraint_language/constants.py

@@ -51,7 +51,7 @@ def __init__(
     def global_params(
         self,
         unit=0.5,
-        segment_margin=1.4,
+        segment_margin=1.8,
         wall_thickness=("uniform", 0.2, 0.3),
         wall_height=("uniform", 2.8, 3.2),
     ):
@@ -65,7 +65,7 @@ def global_params(
         }
 
     def door_params(
-        self, door_width_ratio=("uniform", 0.7, 0.8), door_size=("uniform", 2.0, 2.4)
+        self, door_width_ratio=("uniform", 0.9, 0.95), door_size=("uniform", 2.2, 2.6)
     ):
         door_width = (self.segment_margin - self.wall_thickness) * rg(door_width_ratio)
         assert door_width > 0

infinigen/datagen/configs/compute_platform/local_256GB.gin

@@ -6,7 +6,7 @@ get_cmd.process_niceness = 20 # let UI processes etc take precedence, to make th
 local_submit_cmd.use_scheduler = True
 LocalScheduleHandler.jobs_per_gpu = 1
 jobs_to_launch_next.max_queued_total = 1
-jobs_to_launch_next.max_stuck_at_task = 4
+jobs_to_launch_next.max_stuck_at_task = 12
 
 # All will run locally, LocalScheduleHandler doesnt actually enforce cpu/ram constraints currently
 queue_coarse.submit_cmd = @local_submit_cmd

infinigen_examples/configs_indoor/__init__.py

@@ -0,0 +1 @@
+GraphMaker.room_children='studio'