xsl/hair
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
33 Commits 4 Branches 0 Tags
0e71830bb65d6a9143c7dc46aa4006cc68c5f5a6
Commit Graph
33 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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
Ubuntu 795de68985 save code 2026-06-15 10:33:15 +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
Ubuntu cd4ca278d4 完成接口1 和接口2，测试页面也完成 2026-06-15 10:10:52 +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 94ad95850e feat(接口2): 新增生发后图片(ComfyUI/Flux inpaint) 
在发际线预览基础上，每种发际线再出一张「植发3个月」生发图：

- hairline/mask.py: headmark 5步法遮罩(额头上部区域∩SegFormer头部=ROI，
  取发际线曲线以上闭合区域)；用 hairline_texture_black 渲染黑线替代手绘检测；
  compose_comfy_rgba 合成 RGBA(alpha=255-mask, 透明=重绘区, 对齐 ComfyUI mask=1-alpha)
- hairline/comfyui.py: ComfyUI 客户端(默认8182)，/upload/image+/prompt(改节点26+随机seed)
  +轮询/history+/view 取回生发图
- hairline/render.py: 抽出 build_overlay_layer 供遮罩取曲线像素
- hairline/service.py: extract_context 一次出 landmarks/parse_map/502点；
  generate_grow_results 每种=预览+生发图(同步串行N张，单张ComfyUI失败则grown置空不拖垮整请求)
- app.py: /hair/grow 返回 results[].grown_image_base64；重活放线程池避免卡事件循环
- add_hair.json 工作流 + hairline_texture_black/ 黑贴图入库
- 测试: test_mask.py(遮罩几何) + test_api mock ComfyUI 验 grown 字段，35 全绿

实测(5090): female 5张生发图同步约18s；预览/生发图人物五官服饰背景保持、黑线已清除。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 23:42:30 +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
xslandClaude Opus 4.8 554b64a916 feat(接口2): C端生发发际线预览(真实实现，替换Mock) 
第一步：按性别把发际线类型贴图渲染到照片，输出 N 张发际线叠加预览图。

- hairline/render.py: 解析 face_ext.obj(502 UV + 64 ribbon扩展面) + OpenCV 逐三角
  仿射 warp 渲染器；关键修复——face_ext.obj 是 OBJ序，用 INDEX_MAP_468 把 MP序
  502点重排成 OBJ序后再投影，否则 ribbon 会错贴到中脸
- hairline/service.py: FaceLandmarker+SegFormer 单例 + 性别贴图映射(扫描去空格)
  + generate_previews 管线(female5/male4)
- 集成点修复: face_landmarks DEFAULT_MODEL_PATH 改 hairline/models/;
  constants HF_FACE_PARSER_MODEL 改本地路径(离线)
- app.py: /api/v1/hair/grow 接真实实现，gender 必填(非法→1004)，返回
  results[].image_base64(不落盘)，校验/鉴权同接口1；lifespan 预热接口2单例；
  补 logging.basicConfig
- 依赖: transformers==4.45.2；SegFormer 权重走 hf-mirror 下载(见 OFFLINE_ASSETS)
- 测试: tests/test_hairline.py(mesh/重排/贴图映射) + test_api 接口2用例，31 全绿

注：SegFormer 受 5090/torch 限制走 CPU(~2.5s/张)，换 cu128 可 SEG_DEVICE=cuda。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 20:31:41 +08:00
xslandClaude Opus 4.8 891bc0da8b feat(annotation): 线名移到右侧 + 线条只覆盖人脸范围 
- 5 条横线的线名(头顶/发际线/眉心/鼻翼下缘/下巴尖)从左侧移到右侧
- 四庭 cm(顶庭/上庭/中庭/下庭)留在左侧，右对齐贴脸盒左缘
- 横线只覆盖脸宽(左右脸颊)、竖线只覆盖脸高(头顶→下巴)，渐变消失
- 七眼段宽标注移到脸盒上端/下端(配合新线长)，相邻段上下错行防重叠

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 19:58:12 +08:00
xslandClaude Opus 4.8 10488171eb feat(annotation): 标注图改版——横线标名+四庭cm，竖线标七眼段宽(顶/底) 
按需求重排标注图：
- 去掉右侧全脸总高标尺，去掉单眼宽度/两眼间距/脸宽 三条横向标注
- 横向 5 条线标注线名(头顶/发际线/眉心/鼻翼下缘/下巴尖) + 保留左侧四庭 cm
- 新增纵向 6 条渐变竖线(左脸颊/左右眼内外角/右脸颊)，切脸宽为 5 段
- 每段宽度在顶部和底部各标一次(只标 X.XXcm)，相邻段上下错行防重叠
- 新增 draw_gradient_vertical_line 竖向渐变线
- test_api 用模块级 ACCEPT_PASSWORDS 固定测试密码，不依赖 worker_config.json

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 18:37:09 +08:00
xsl 9391408088 Merge branch 'main' of http://git.xiangsilian.com:3000/xsl/hair 2026-06-14 18:23:25 +08:00
Ubuntu eaafbc97ea save code 2026-06-14 18:23:08 +08:00
xslandClaude Opus 4.8 023bb2e6fd feat(worker): start.sh 改为后台服务开关(start/stop/restart/status) 
- ./start.sh 现支持 start/stop/restart/status 子命令，无参默认 start
- 后台运行 + PID 文件(worker.pid) + 日志(worker.log)，手动控制开关
- 前台热重载开发仍用 run_worker.sh
- gitignore 补 worker.pid / worker.log

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 17:47:49 +08:00
xslandClaude Opus 4.8 0eaaf05ac3 fix(annotation): 标注图重排，8 个量全标注且不重叠 
- 修复七眼三个横向量(单眼宽度/两眼间距/脸宽)叠在同一条眼睛线上的重叠问题，
  改为上下错开三个高度，各自虚线带箭头 + 紧贴标签
- 新增「全脸总高」标注：右侧竖向标尺(虚线双箭头，头顶→下巴)+ 标签
- 四庭(顶/上/中/下庭)仍在左侧各段中点
- 标签用全名：单眼宽度/两眼间距/脸宽/全脸总高
- 新增 tests/test_annotation.py 防回归

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-14 17:33:58 +08:00
xsl 78226ae0b9 chore: SegFormer 模型不入库，改为手动下载 
- .gitignore 排除 hairline/models/face-parsing/model.safetensors（~323MB）
- OFFLINE_ASSETS.md 更新：区分已入库文件和需手动下载文件，补充 SegFormer sha256
 2026-06-14 17:19:24 +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
xsl ad1b95df4e Merge branch 'main' of http://git.xiangsilian.com:3000/xsl/hair 2026-06-14 16:49: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 5e513c7006 添加发型图片 2026-06-14 15:31:29 +08:00
Ubuntu 3b706ec0ef 网关开发完成 2026-06-14 15:04:34 +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 7493f48992 加入模型文件 2026-06-14 00:24:09 +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
UbuntuandCursor 5313206976 初始化 FastAPI HTTPS Hello World 服务 
包含 FastAPI 应用、Nginx 反向代理配置、systemd 服务及腾讯云 pip 源配置。

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-06-04 21:59:53 +08:00