chore(layout): misc update for layout sim. (#43)

embodied_gen/scripts/gen_layout.py

@@ -115,7 +115,9 @@ def entrypoint() -> None:
         # Background GEN (for efficiency, temp use retrieval instead)
         bg_node = layout_info.relation[Scene3DItemEnum.BACKGROUND.value]
         text = layout_info.objs_desc[bg_node]
-        match_key = SCENE_MATCHER.query(text, str(scene_dict))
+        match_key = SCENE_MATCHER.query(
+            text, str(scene_dict), params=gpt_params
+        )
         match_scene_path = f"{os.path.dirname(args.bg_list)}/{match_key}"
         bg_save_dir = os.path.join(output_root, "background")
         copytree(match_scene_path, bg_save_dir, dirs_exist_ok=True)

embodied_gen/utils/geometry.py

@@ -137,7 +137,7 @@ def with_seed(seed_attr_name: str = "seed"):
 def compute_convex_hull_path(
     vertices: np.ndarray,
     z_threshold: float = 0.05,
-    interp_per_edge: int = 3,
+    interp_per_edge: int = 10,
     margin: float = -0.02,
     x_axis: int = 0,
     y_axis: int = 1,

embodied_gen/validators/quality_checkers.py

@@ -574,7 +574,7 @@ def test_semantic_matcher(
     #     "presence_penalty": 0.3,
     # }
     gpt_params = None
-    match_key = SCENE_MATCHER.query(text, str(scene_dict))
+    match_key = SCENE_MATCHER.query(text, str(scene_dict), params=gpt_params)
     print(match_key, ",", scene_dict[match_key])
 
 

tests/test_integration/test_pipeline.sh

@@ -9,8 +9,8 @@ text3d-cli --prompts "small bronze figurine of a lion" \
     --n_image_retry 2 --n_asset_retry 2 --n_pipe_retry 1 --seed_img 0 \
     --output_root ${output_dir}/textto3d
 
-texture-cli --mesh_path "apps/assets/example_texture/meshes/robot_text.obj" \
---prompt "举着牌子的写实风格机器人，大眼睛，牌子上写着“Hello”的文字" \
+texture-cli --mesh_path "apps/assets/example_texture/meshes/horse.obj" \
+--prompt "A gray horse head with flying mane and brown eyes" \
 --output_root "${output_dir}/texture_gen" \
 --seed 0