diff --git a/embodied_gen/scripts/gen_layout.py b/embodied_gen/scripts/gen_layout.py index 8f183a0..c78b0df 100644 --- a/embodied_gen/scripts/gen_layout.py +++ b/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) diff --git a/embodied_gen/utils/geometry.py b/embodied_gen/utils/geometry.py index 7623107..8352ccc 100644 --- a/embodied_gen/utils/geometry.py +++ b/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, diff --git a/embodied_gen/validators/quality_checkers.py b/embodied_gen/validators/quality_checkers.py index e289310..378e3af 100644 --- a/embodied_gen/validators/quality_checkers.py +++ b/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]) diff --git a/tests/test_integration/test_pipeline.sh b/tests/test_integration/test_pipeline.sh index 7135723..f6a395b 100644 --- a/tests/test_integration/test_pipeline.sh +++ b/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