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
25 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 a86125246e docs: 网关侧改动清单(JPG落盘扩展名嗅探等) 
汇总 worker 近期变更里与网关有关的点供网关侧应用：
- [功能必需] base64→url 落盘按内容嗅探扩展名(PNG/JPG)，因接口2/3/5 改 JPG(已在 forward.py 改)
- [建议] 生发接口超时≥120s
- [确认] 递归改写覆盖数组内字段/可空 null
- [可选] OpenAPI 表单 gender/grow-b 声明
- 已完成项：接口4 网关实现/200状态/接口3去original

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 23:44:41 +08:00
xslandClaude Opus 4.8 4c7681b338 perf(图片): 接口2/3/5 返回 JPG(体积~9×↓)，接口1 标注图仍 PNG(透明) 
- app.py: 接口2(预览+生发)/3(生发)/5(发际线叠图) 编码改 JPG(质量90,env JPG_QUALITY);
  接口1 annotated_image 含透明仍 PNG。_png_to_jpg_b64 把 ComfyUI 的 PNG 重编码为 JPG(无法解码则透传)
- gateway/forward.py: 落盘按内容嗅探扩展名(PNG头→.png 否则.jpg)，原先硬编码 .png
- 测试/文档同步；实测接口5 一张 59KB(JPG) vs 548KB(PNG)。pytest 44 全绿

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 23:35:24 +08:00
xslandClaude Opus 4.8 e3c67fc8cf feat(comfyui): ComfyUI 改 8188 + HTTP Basic Auth 
本机 ComfyUI 开了 Basic Auth(user admin)，端口 8182→8188：
- hairline/comfyui.py: 默认 URL 8188；所有 httpx 请求带 auth=(user,password)
  密码来源 环境变量 COMFYUI_PASSWORD → worker_config.json.comfyui_password → password.txt
- password.txt 入 .gitignore(含密码不入 git)；worker_config.example 加 comfyui_password
- 实现说明文档同步(8188 + Basic Auth)
实测：ping + 实跑一张生发图均通过(带鉴权)。pytest 44 全绿。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 23:10:09 +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 043a4c0603 docs: 接口4 网关侧实现方案(供网关开发照做) 
接口4 不碰本地GPU/模型，仅调外网豆包→放网关本地实现更合理。文档含：
路由改本地处理(不转发)、httpx调方舟(OpenAI兼容,无需SDK)、prompt原文、
base64 data URI喂图、字段映射(6英文+全部中文)、无人脸→1001、配置(ark密钥入网关config)、
错误码、worker侧回收步骤、自测。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 20:33:50 +08:00
xslandClaude Opus 4.8 3008552331 docs(网关): 接口4 已实现说明(无图片字段透传+外网豆包调用耗时) 
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 10:51:02 +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 147fef6ca6 docs(网关): 补全 base64→url 映射表(接口1/2/3/5) + 嵌套数组/超时提醒 
接口 1/2/3/5 已真实实现，补齐网关需要的字段映射，供网关开发参考：
- 网关任务书 §6：完整映射表(annotated_image / results[].image / results[].grown_image /
  best_hairline_image / hair_growth_image / hairline_images[].image)；推荐"凡 *_base64 递归改写"
  通用实现；可空字段(生发图)保留 null；gender 等入参网关透传无需改造
- 网关任务书 §9：接口4 仍 mock；生发接口(2/3) ComfyUI 同步出图慢，request_timeout 调大≥120s
- 架构 §9：标注两个易漏点(数组内字段需递归、生发图可空)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 00:29:33 +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 ce95a508c1 feat(接口3): B端生发-马克笔发际线检测+生发(替换Mock) 
医生在额头用马克笔画规划发际线 → 检测该线 → 生发。检测算法源自 /home/xsl/headmark。

- hairline/marker_detect.py: 黑帽响应图(MORPH_BLACKHAT)+鬓角锚点(MediaPipe 21/251吸附)
  +skimage route_through_array 最小路径检测画线；路径平均响应阈值拒识无画线
  (headmark 调研：全局灰度阈值不可用，黑帽+Dijkstra 实测误差≤0.5px)
- hairline/mask.py: 抽出 mask_from_curve(曲线+ROI闭合)，接口2/3共用
- hairline/service.py: generate_grow_b——检测→遮罩→原图重画干净线→ComfyUI生发
- app.py: /hair/grow-b 真实实现，marked+original各三选一+校验；输出
  best_hairline_image_base64(=原图)/hair_growth_image_base64/hairline_type="custom"；
  无人脸或未检测到画线→1001；重活进线程池
- requirements: scikit-image==0.24.0 (⚠️锁0.24，0.25+强依赖numpy>=2会顶掉mediapipe的numpy<2)
- 文档: docs/接口3-B端生发-技术实现方案.md
- 测试: test_marker.py(检测/拒识/辅助) + test_api grow-b(mock ComfyUI)，42全绿

实测(5090): grow-b ~6.4s，生发图把额头发际线补到医生画线、清除划线、人物保持。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-15 00:08:05 +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 e827b23c1d Merge branch 'main' of http://git.xiangsilian.com:3000/xsl/hair 2026-06-14 17:08:16 +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
xslandClaude Opus 4.8 8d3b145111 feat(worker): 接口1 四庭七眼测量真实实现（替换 Mock） 
worker 侧从 Mock 替换为真实算法：
- face_analysis 包：detector(MediaPipe 478点) / pose(solvePnP 姿态) /
  calibration(虹膜直径法) / hair_segmenter+bisenet_model(方案B 头发分割) /
  measure(方案A兜底+B/A决策+七眼+换算) / annotation(numpy渐变线+中文标注)
- app.py：/api/v1/face/measure 接真实实现，返回 annotated_image_base64
  （不落盘不拼URL，落盘由网关做）；加 X-Internal-Token 鉴权、/health 就绪态、
  可配置分辨率门槛、异常兜底
- 部署：start.sh/run_worker.sh/hair-worker.service 监听 8187；worker_config 示例
- 测试 tests/：Tier1合成真值<1e-6 + Tier2缩放不变 + Tier3叠加 + 错误码集成 +
  数值回归，pytest 24 项全绿
- 文档补实测基线表 + RTX5090/torch 说明

注：worker 为 RTX 5090(sm_120)，pinned torch 2.2.2(cu121) 只到 sm_90，
BiSeNet 已自动回退 CPU（方案B 正常）；要用 GPU 需换 torch cu128(≥2.7)。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 16:07:28 +08:00
xsl bf68a32d9f docs: 拆分网关与worker文档，按机器分别开发 
- 新增 网关-开发任务书.md: 独立自包含的网关侧任务书(骨架/健康检查派发/
  鉴权转发/base64改URL/静态托管/部署/DoD)，在外网机开发
- worker任务书 v1.5: 移除§16网关章节，顶部标注本机职责，DoD只留worker侧，
  指向独立网关任务书
- 新增 docs/README.md: 文档索引，标清worker机/网关机/共享分别读哪些
 2026-06-14 14:29:48 +08:00
xsl 2e789d4efa docs: 架构拆分为外网网关+高性能worker(GPU) 
- 新增 系统架构-网关与高性能后端.md: 网关薄代理(健康检查/空闲派发/鉴权/
  base64落盘改URL/无后端→1007) + worker(GPU跑完整app)职责划分、配置文件、
  鉴权(共享密码可轮换)、标注图base64流程、部署、安全注意
- 关键约束: 接口文档不变; 每worker并发=1, 总并发=健康worker数; 多worker可配置
- 技术方案: 加运行位置横幅; torch改GPU(CUDA); handler返回annotated_image_base64
- 任务书v1.4: 标注差异说明; 阶段八返base64+鉴权中间件; 阶段十worker(GPU)部署;
  新增§16网关工作流; DoD拆分worker侧/网关侧
- OFFLINE_ASSETS: 区分worker(GPU torch)与网关(轻量)依赖
 2026-06-14 13:55:45 +08:00
xsl 66a2c84168 chore: 预下载离线模型权重+字体，补充内网部署清单 
- 下载 BiSeNet(79999_iter.pth 53MB) + resnet18骨干(45MB) + Noto Sans CJK SC字体(16MB)
  到 face_analysis/weights 与 fonts（大文件 .gitignore 不入库，靠文件拷贝带入内网）
- 新增 OFFLINE_ASSETS.md：文件清单+sha256校验+resnet18骨干离线放置说明+来源URL
- 字体由 SourceHanSansSC 改为 NotoSansCJKsc(同一套字体)，同步更新两份文档引用
- 文档补充内网/离线注意事项(resnet18联网下载坑、权重已预置)
 2026-06-14 00:23:10 +08:00
xsl 9d2ff0b4b2 docs: 接口1四庭七眼 技术方案v2.0 + 开发任务书 + 测试夹具 
- 技术方案修订v2.0: 引入方案B(BiSeNet分割取真实发际线/头顶)解决方案A循环论证;
  修复分辨率写反/PingFang字体/numpy2.x冲突/渐变线性能;solvePnP替换正面判定;
  分辨率门槛放宽至短边600长边800且可配置
- 新增开发任务书(给AI agent执行): 10阶段串行步骤+交付物+验证方法+DoD;
  §14测试素材清单 §15三层精度验证策略(合成真值/缩放不变性/可视化)
- tests/fixtures: frontal/hard_longhair/lowres/landscape/corrupt 5个夹具
  (1006超大图改测试时动态生成不入库)
 2026-06-13 23:56:34 +08:00
Ubuntu 2c669fa7e5 save code 2026-06-13 23:04:37 +08:00
Ubuntu d8a6e53979 save code 2026-06-13 12:11:31 +08:00
xsl 1233952571 save code 2026-06-04 22:05:26 +08:00