xsl/hair
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
47 Commits 4 Branches 0 Tags
5311c8d7c7c124b26142ffb9f04a4401bf9a50b7
Commit Graph
10 Commits
Author SHA1 Message Date
xslandClaude Opus 4.8 5311c8d7c7 feat(接口2/3): 加 use_mask 开关用于遮罩对比；网关回退纯透传 
- 接口3 generate_grow_b 增加 use_mask（默认 true）：false 跳过检测、
  直接送划线图（空遮罩）
- 接口2 generate_grow_results 增加 use_mask：false 用干净原图+空遮罩，
  只跑一次 ComfyUI、N 项复用
- app.py 接口2/3 增加 use_mask form 参数并透传
- gateway/app.py 回退为纯透传（移除 OpenAPI schema 注入，请求体原样转发）
- docs 补 use_mask 说明

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-17 23:01:12 +08:00
xsl 52f1bb4c0b 修改接口3 加入是否用遮罩接口 2026-06-17 22:27:06 +08:00
xslandClaude Opus 4.8 76f7c06905 review+docs: 接口4 回收到网关、对齐错误码、合并文档为一份实现说明 
代码 review 后的清理：
- 接口4 由网关本机实现，worker app.py 的 /face/features 回退 Mock(保持 worker 无外网依赖)；
  worker requirements 标注 volcengine 改为网关侧；移除 worker 的接口4 测试(随实现挪到网关)
- 网关接口4 业务错误 HTTP 状态统一改 200(与其余接口/worker 约定一致，原为400/503)
- 接口文档：gender 非法码 1004(原误写1008)；修正指向已删文档的链接

文档合并：把各接口技术方案/开发任务书/系统架构/网关任务书 合并成 docs/实现说明.md(简要总览)，
删除原 7 份分散文档，README 收敛为索引(实现说明/接口文档/需求/OFFLINE_ASSETS)。

pytest 44 全绿。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 20:58:33 +08:00
xslandClaude Opus 4.8 0e71830bb6 feat(接口4): 用户特征-火山方舟豆包视觉模型(替换Mock) 
接口4 改用远程视觉大模型(算法来源 /home/xsl/fuyan)，一次返回几十项面部特征。

- face_features.py: 火山方舟 Ark 客户端(单例)+doubao-seed-1-6-vision prompt(移植fuyan)
  +base64 data URI喂图(实测doubao接受)+解析JSON+映射6英文优先字段(face_shape等)
  并保留doubao全部中文字段；has_face 据"图片是否有人脸"判定
- app.py: /face/features 真实实现，三选一图(image_url直传doubao,file/base64转dataURI)，
  features 为 JSON 字符串；无人脸→1001；重活线程池
- 配置: API Key 走 worker_config.json.ark_api_key / 环境变量 ARK_API_KEY(不入git);
  worker_config.example 加占位; requirements 加 volcengine-python-sdk[ark]
- 文档: 接口文档/OpenAPI 更新为豆包实现+几十项字段+1001
- 测试: mock doubao 的成功/无人脸/多参，47全绿

⚠️ 唯一调外网云模型的接口，worker 需可达 ark.cn-beijing.volces.com(已实测可达)。
实测 frontal: 42字段, 鹅蛋脸/平眉/18-25岁/静态型/女/少年型。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 10:50:17 +08:00
xslandClaude Opus 4.8 bf76923591 refactor(接口3): 简化为只需划线图一张，去掉 original + best_hairline 
按需求方意见——B端只需上传一张已画好发际线的图，用不着原图：
- 入参去掉 original_image_*，只保留 marked_image_*(三选一)
- 输出去掉 best_hairline_image_url，只返回 hair_growth_image_url + hairline_type
- ComfyUI 输入图改用 marked 划线图原样(add_hair.json 本就是"画了线的照片"，
  提示词清除黑线)；检测路径只用于建遮罩，不再重画干净线/不需对齐原图
- service.generate_grow_b 签名改 (marked_bgr) 单参
- 同步文档：接口文档/接口3技术方案/网关映射表(去掉接口3 best_hairline 行)
- 测试更新：grow-b 只传 marked，断言无 best_hairline 字段，44全绿

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 10:28:13 +08:00
xslandClaude Opus 4.8 d4e6a794d8 docs(接口2): 标注生发图已实现，清理"只做预览"过期文案 
接口2 的生发后图片(§10 ComfyUI/Flux)已实现，更新早期"本期只做预览/不做生发"的描述：
- 技术方案：标题/§0/§9.6 改为"预览+生发两步均已实现"，§9.6 列出后续优化方向
- 接口文档：接口2 说明改为"每方案返回 image_url(预览)+grown_image_url(生发图)"

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 09:47:42 +08:00
xslandClaude Opus 4.8 38161d1b50 feat(接口5): 发际线PNG生成(真实实现，替换Mock) 
复用接口2 预览管线：gender 必填 → 该性别全部发际线叠加图(同接口2预览) +
最佳(order=1)发际线曲线的面部中间点坐标。无生发(不调 ComfyUI)。

- hairline/service.py: generate_hairline_pngs——N张发际线叠图 +
  best_center(面部中轴眉心x × order1曲线在该处的y)
- app.py: /hairline/generate 真实实现，新增 gender 必填(非法→1004)，
  返回 hairline_images[].image_base64 + best_hairline_center_point；
  无人脸→1001；重活线程池
- 接口文档/OpenAPI: 接口5 新增 gender 入参 + 输出改 base64(网关改url)
- 测试: test_api 接口5(gender必填/N张/center点)，44全绿

实测(5090): female 5张叠图 + center{x,y}，~2.5s(无Flux)。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 00:20:56 +08:00
xslandClaude Opus 4.8 4c69bb4623 docs(接口2): 新增生发图设计(§10 ComfyUI+Flux) + headmark遮罩算法 
- 技术方案 §10：生发图生成管线（add_hair.json 工作流解读、遮罩算法、
  ComfyUI 客户端、契约变更、M5~M8 步骤、风险）
- 遮罩算法参考 /home/xsl/headmark 5步法，用 hairline_texture_black 渲染黑线
  替代手绘检测：额头上部区域 ∩ 头部分割 = ROI，取发际线曲线以上闭合区域
- 接口文档：results[] 新增 grown_image_url(生发后图片) + 同步/超时说明

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 23:32:21 +08:00
xsl db32fa12e5 feat(接口2): 移植head3d发际线管线 + 接口2实现方案文档 
- 从head3d复制发际线检测管线到 hairline/ 包：MediaPipe Tasks + SegFormer分割
  + 17锚点射线检测 + 502点mesh(face_ext.obj)+UV
- 复制模型：face_landmarker.task(3.7MB)、SegFormer config/preprocessor
  (model.safetensors 340MB 单独下载中)
- 新增 docs/接口2-C端生发-技术实现方案.md：第一步=发际线曲线叠加预览图，
  新增gender必填参数，按性别贴图数量输出(female5/male4)，hairline_type英文key，
  服务端cv2逐三角形warp渲染器(head3d只有浏览器端Three.js渲染)
- 接口文档.md 接口2章节同步：gender参数、输出语义、错误码说明
- hairline_texture/ 9张发际线贴图入库
 2026-06-14 16:59:39 +08:00
Ubuntu d8a6e53979 save code 2026-06-13 12:11:31 +08:00