优化为4b 模型

face_analysis/hairline_grow.py

@@ -110,7 +110,7 @@ def _gray_b64(gray_float):
 def _red_mask_b64(mask_bool, h, w):
     """布尔遮罩 → 纯红 alpha PNG data URI。
     遮罩区域 RGBA=(255,0,0,255)，其余区域 RGBA=(0,0,0,0)。
-    供外部 ComfyUI 重绘服务（如 local_test）按 alpha 通道识别重绘区。
+    供 ComfyUI 重绘接口（/api/v1/redraw）按 alpha 通道识别重绘区。
 
     注意：cv2.imencode 写 PNG 用的是 **BGRA** 顺序（B,G,R,A），所以要得到
     浏览器显示的红色 R=255，需赋值 (B=0,G=0,R=255,A=255)。
@@ -357,7 +357,8 @@ def _redraw_band_mask(inner_pts, outer_pts, h, w, rid="", upper=None,
 
 
 def compute_mask(image_bgr, landmarks, seg_model, mask_type, erode_cm, px_per_cm,
-                 hairline_push_cm=0.0, hairline_edge="column", rid="", render_viz=True):
+                 hairline_push_cm=0.0, hairline_edge="column", rid="", render_viz=True,
+                 hair_mask=None):
     """算出布尔遮罩 + 可视化。
 
     seg_model: bisenet | segformer。
@@ -368,6 +369,7 @@ def compute_mask(image_bgr, landmarks, seg_model, mask_type, erode_cm, px_per_cm
     render_viz: 是否生成各阶段叠图 overlay JPG（接口11 调试页用）。接口2/12 路径传 False
         可跳过 6+ 张 base64 编码，省 ~80ms；数据字段（_inner_pts/_outer_pts/_upper_mask/
         mask_pixels 等）始终返回，不受影响。
+    hair_mask: 预计算的头发布尔遮罩（来自 SegFormer parse）。传入时跳过重复分割，省 ~0.9s。
     返回 (mask_bool, viz_dict)。
     """
     lg = lambda msg: logger.info("[%s] %s", rid, msg) if rid else None
@@ -381,13 +383,16 @@ def compute_mask(image_bgr, landmarks, seg_model, mask_type, erode_cm, px_per_cm
     upper = _upper_region_mask(baseline_pts, w, h)
     lg(f"baseline 第一点={baseline_pts[0]} 末点={baseline_pts[-1]} upper像素={int(upper.sum())}")
 
-    if seg_model == "bisenet":
-        hair_mask = _bisenet_hair_mask(image_bgr, landmarks, w, h)
-    elif seg_model == "segformer":
-        hair_mask = _segformer_hair_mask(image_bgr)
+    if hair_mask is None:
+        if seg_model == "bisenet":
+            hair_mask = _bisenet_hair_mask(image_bgr, landmarks, w, h)
+        elif seg_model == "segformer":
+            hair_mask = _segformer_hair_mask(image_bgr)
+        else:
+            raise ValueError(f"未知 seg_model: {seg_model}")
+        lg(f"头发分割完成 seg_model={seg_model} hair_pixels={int(hair_mask.sum())}")
     else:
-        raise ValueError(f"未知 seg_model: {seg_model}")
-    lg(f"头发分割完成 seg_model={seg_model} hair_pixels={int(hair_mask.sum())}")
+        lg(f"头发分割跳过(复用外部传入) hair_pixels={int(hair_mask.sum())}")
 
     top_fill = _fill_to_baseline(hair_mask, upper)      # 含额头，延伸到图底
     closed = _largest_cc(top_fill & upper)              # 闭合区域：头发+额头，底=基线
@@ -849,7 +854,8 @@ def _grow_core(image_bgr, hairline_id, *, is_hr, seg_model, erode_cm, swap_mode,
                edge_erode_px, denoising_strength, gen_backend, hairgrow_strength,
                mb_levels, hairline_push_cm, hairline_edge, blend_method, color_match,
                color_match_strength, mb_feather_px, transition_band_px,
-               inpainting_fill, mask_blur, mask_dilate_scale, rid, render_viz=True):
+               inpainting_fill, mask_blur, mask_dilate_scale, rid, render_viz=True,
+               hair_mask=None):
     """接口11 共享核心：遮罩(pushed)→生成→硬贴回→接缝融合，产出 ④ final。
 
     不做任何重绘。返回中间产物 dict（供接口11 构造响应、接口12 取 final+重绘带用）：
@@ -879,7 +885,7 @@ def _grow_core(image_bgr, hairline_id, *, is_hr, seg_model, erode_cm, swap_mode,
     mask_bool, mask_viz = compute_mask(
         image_bgr, landmarks, seg_model, mask_type, erode_cm, px_per_cm,
         hairline_push_cm=hairline_push_cm, hairline_edge=hairline_edge, rid=rid,
-        render_viz=render_viz)
+        render_viz=render_viz, hair_mask=hair_mask)
     t_mask = time.time() - t0
     logger.info("[%s] 步骤1 遮罩完成 耗时=%dms mask_pixels=%d", rid, int(t_mask*1000), int(mask_bool.sum()))
 
@@ -1028,13 +1034,14 @@ def generate_hairline_redraw(image_bgr, hairline_id, is_hr=False, seg_model="seg
                              transition_band_px=-1,
                              inpainting_fill=1, mask_blur=11, mask_dilate_scale=1.0,
                              comfyui_prompt=None, beauty_alpha=0.6,
-                             band_lo_mult=0.5, band_hi_mult=1.5, rid=None):
+                             band_lo_mult=0.5, band_hi_mult=1.5, rid=None,
+                             hair_mask=None):
     """接口12 发际线带重绘。内部先跑接口11 核心拿到 ④ final，再取 ⑤-① 发际线重绘带
     （外推↔内推之间、经 baseline 截断只留上部）作遮罩。
 
     **本接口不再做 Flux-2 重绘**：只产出 `final`（接缝融合基底）+ 纯红遮罩
-    `redraw_band_mask`（RGBA，遮罩区=(255,0,0,255)、其余全透明），重绘交给前端调
-    外部 ComfyUI 重绘服务（见 local_test）完成。旧的 `redraw_full` / `redraw_band`
+    `redraw_band_mask`（RGBA，遮罩区=(255,0,0,255)、其余全透明），重绘交给后端
+    ComfyUI 重绘接口（/api/v1/redraw）完成。旧的 `redraw_full` / `redraw_band`
     字段保留为空，仅作结构兼容。
 
     返回可直接进 ok() 的 data dict。未检出人脸抛 NoFaceError。
@@ -1056,7 +1063,8 @@ def generate_hairline_redraw(image_bgr, hairline_id, is_hr=False, seg_model="seg
         color_match=color_match, color_match_strength=color_match_strength,
         mb_feather_px=mb_feather_px, transition_band_px=transition_band_px,
         inpainting_fill=inpainting_fill, mask_blur=mask_blur,
-        mask_dilate_scale=mask_dilate_scale, rid=rid, render_viz=False)
+        mask_dilate_scale=mask_dilate_scale, rid=rid, render_viz=False,
+        hair_mask=hair_mask)
     final = core["final"]
     mask_viz = core["mask_viz"]
     w, h = core["w"], core["h"]
@@ -1089,7 +1097,7 @@ def generate_hairline_redraw(image_bgr, hairline_id, is_hr=False, seg_model="seg
         redraw_info = {"enabled": False, "error": f"band: {ex}"}
 
     # ② Flux-2 重绘已下线：本接口现在只产出 final（接缝融合基底）+ 纯红重绘带遮罩，
-    #    重绘交给前端调外部 ComfyUI 重绘服务（见 local_test）完成。
+    #    重绘交给后端 ComfyUI 重绘接口（/api/v1/redraw）完成。
     #    下面保留 redraw_full_b64 / redraw_band_b64 为空，保持返回结构兼容（旧字段）。
     redraw_full_b64 = ""
     redraw_band_b64 = ""