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

This commit is contained in:
xsl
2026-06-17 22:27:06 +08:00
parent 01397c414e
commit 52f1bb4c0b
5 changed files with 101 additions and 27 deletions
+2 -1
View File
@@ -594,6 +594,7 @@ async def hair_grow_b(
marked_image_file: Optional[UploadFile] = File(default=None, description="划线图片文件(JPG/PNG,≤ 1 MB"),
marked_image_url: Optional[str] = Form(default=None, description="划线图片 URL"),
marked_image_base64: Optional[str] = Form(default=None, description="划线图片 base64"),
use_mask: bool = Form(default=True, description="是否启用自动检测遮罩(测试对比用)。false 时跳过检测、直接送划线图(空遮罩)"),
):
# 划线图三选一取图(只需这一张)
marked_raw, e = await resolve_image_bytes(marked_image_file, marked_image_url, marked_image_base64)
@@ -615,7 +616,7 @@ async def hair_grow_b(
from fastapi.concurrency import run_in_threadpool
from hairline.service import generate_grow_b
res = await run_in_threadpool(generate_grow_b, marked)
res = await run_in_threadpool(generate_grow_b, marked, use_mask)
if res["status"] == "no_face":
return err(1001, "无法识别人像")
if res["status"] == "no_line":