修改接口3 加入是否用遮罩接口

hairline/service.py

@@ -193,26 +193,33 @@ def generate_hairline_pngs(image_bgr: np.ndarray, gender: str):
     return {"images": images, "best_center": best_center}
 
 
-def generate_grow_b(marked_bgr: np.ndarray):
+def generate_grow_b(marked_bgr: np.ndarray, use_mask: bool = True):
     """接口3：检测医生手绘发际线 → 遮罩 → 送 ComfyUI 生发（仅需划线图一张）。
 
     检测路径只用来**建遮罩**；ComfyUI 输入图用 **marked 原图**（含医生手绘线，
     工作流提示词会清除黑线再生发）。
+
+    use_mask（默认 True）：是否启用自动检测的遮罩，用于测试对比。
+      - True：检测手绘线 → 建遮罩 → alpha=255−mask（透明区=重绘区，节点44 画黄色参考区）。
+      - False：跳过检测，直接送划线图，alpha 全 255（空遮罩，节点26 mask 为空），
+        模型仅凭医生黑线参考生发。无需改工作流，唯一变量是遮罩。
     Returns: {"grown_png": bytes 或 None, "status": "ok"|"no_face"|"no_line"}。
     """
-    rgb = cv2.cvtColor(marked_bgr, cv2.COLOR_BGR2RGB)
-    landmarks = get_landmarker().detect(rgb)
-    if landmarks is None:
-        return {"grown_png": None, "status": "no_face"}
-    parse_map = get_parser().parse(rgb)
-    path = detect_marker_hairline(marked_bgr, landmarks, parse_map)
-    if path is None:
-        return {"grown_png": None, "status": "no_line"}
-
     h, w = marked_bgr.shape[:2]
-    line_w = max(2, int(w * 0.006))
-    curve_mask = path_to_curve_mask(path, h, w, thickness=max(3, line_w))
-    mask = mask_from_curve(curve_mask, landmarks, parse_map)
+    if use_mask:
+        rgb = cv2.cvtColor(marked_bgr, cv2.COLOR_BGR2RGB)
+        landmarks = get_landmarker().detect(rgb)
+        if landmarks is None:
+            return {"grown_png": None, "status": "no_face"}
+        parse_map = get_parser().parse(rgb)
+        path = detect_marker_hairline(marked_bgr, landmarks, parse_map)
+        if path is None:
+            return {"grown_png": None, "status": "no_line"}
+        line_w = max(2, int(w * 0.006))
+        curve_mask = path_to_curve_mask(path, h, w, thickness=max(3, line_w))
+        mask = mask_from_curve(curve_mask, landmarks, parse_map)
+    else:
+        mask = np.zeros((h, w), np.uint8)   # 空遮罩：alpha 全 255，跳过检测
 
     buf = io.BytesIO()
     compose_comfy_rgba(marked_bgr, mask).save(buf, format="PNG")   # marked 原图 + 遮罩