liying/embodiedgen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Xinjie 2025-09-28 19:46:18 +08:00 committed by GitHub
parent 137855bdc6
commit f8b1f1635f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
embodied_gen/scripts/gen_layout.py
View File

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

2
embodied_gen/utils/geometry.py
View File

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

2
embodied_gen/validators/quality_checkers.py
View File

@ -574,7 +574,7 @@ def test_semantic_matcher(
# "presence_penalty": 0.3, # "presence_penalty": 0.3,
# } # }
gpt_params = None 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]) print(match_key, ",", scene_dict[match_key])

4
tests/test_integration/test_pipeline.sh
View File

@ -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 \ --n_image_retry 2 --n_asset_retry 2 --n_pipe_retry 1 --seed_img 0 \
--output_root ${output_dir}/textto3d --output_root ${output_dir}/textto3d
texture-cli --mesh_path "apps/assets/example_texture/meshes/robot_text.obj" \ texture-cli --mesh_path "apps/assets/example_texture/meshes/horse.obj" \
--prompt "举着牌子的写实风格机器人，大眼睛，牌子上写着“Hello”的文字" \ --prompt "A gray horse head with flying mane and brown eyes" \
--output_root "${output_dir}/texture_gen" \ --output_root "${output_dir}/texture_gen" \
--seed 0 --seed 0