xsl
a3fddbbc34
按 MediaPipe 左右对应额头点协调模板延伸距离，保留人脸姿态与头部轮廓钳制，统一预览白线和生发遮罩的几何。补充对齐回归测试并更新贴图顺序测试及实现说明。 验证：13 项相关测试通过；两张用户图片完成前后对比，新图男性 7 型接口与生发图均成功返回。
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
"""接口2 单元测试：mesh 解析 / MP→OBJ 重排 / 性别贴图映射（不需 SegFormer）。"""
|
|
import numpy as np
|
|
|
|
from hairline.render import load_ext_mesh, mp_order_to_obj_order
|
|
from hairline.service import get_texture_map
|
|
from hairline import constants as C
|
|
|
|
|
|
def test_ext_mesh():
|
|
uv, ext_faces = load_ext_mesh()
|
|
assert uv.shape == (502, 2)
|
|
assert len(ext_faces) == 64 # ribbon 扩展三角形
|
|
# 扩展面至少含一个 ≥468 的顶点
|
|
assert all(max(f) >= 468 for f in ext_faces)
|
|
|
|
|
|
def test_mp_to_obj_anchor_mapping():
|
|
"""OBJ 序重排后，ribbon 引用的 obj 锚点应映射回 MP_TOP_ANCHORS。"""
|
|
pts = np.zeros((502, 3), np.float32)
|
|
pts[:468, 0] = np.arange(468) # 用 x 编码 MP 索引
|
|
obj = mp_order_to_obj_order(pts)
|
|
_uv, ext_faces = load_ext_mesh()
|
|
obj_anchor_ids = sorted({pi for f in ext_faces for pi in f if pi < 468})
|
|
mapped = sorted(int(obj[i, 0]) for i in obj_anchor_ids) # 还原成 MP 索引
|
|
assert mapped == sorted(C.MP_TOP_ANCHORS)
|
|
|
|
|
|
def test_texture_map():
|
|
m = get_texture_map()
|
|
assert [k for k, _ in m["female"]] == [
|
|
"ellipse", "flower", "heart", "straight", "wave", "bigflower", "clasicalflower",
|
|
]
|
|
assert [k for k, _ in m["male"]] == [
|
|
"ellipse", "inverse_arc", "m", "straight", "heart", "Softpetal", "wave",
|
|
]
|