xsl/hair
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
121 Commits 4 Branches 0 Tags
b61ea6f33bbcff71cb404f5ed60d565cc07c45d8
Commit Graph
31 Commits
Author SHA1 Message Date
xsl b61ea6f33b feat(接口5): 生发机制对齐接口2 + 统一ComfyUI重绘提示词 
接口5改造（生发机制与接口2一致，按性别分流）：
- generate_hairline_pngs: 生发图来源从 _grow_from_texture(局部inpaint)
  改为按性别分流——female走 generate_grow_results_swap(swapHair+Flux整帧重绘)，
  male走 generate_grow_results(ComfyUI add_hair)
- 新增参数 redraw_max_side/unet_name/v2_defaults(female路径透传)
- 接口5 handler 加 flux_model/redraw_max_side Form参数
- 保留接口5独有输出: 3档叠图(middle/high/low)/中心点/face_measure
- 已验证: female日志出现"接口2女 管线降分辨率max_side=640"+swap+Flux;
  male走add_hair.json; generate_grow_image=false正确跳过生发

统一ComfyUI重绘提示词:
- "填充遮罩区域的头发，皮肤加一点磨皮，再加一点美颜" → "填充遮罩区域的头发"
- 覆盖: _REDRAW_PROMPT/_DEFAULT_PROMPT常量 + app.py各接口Form默认 +
  工作流JSON节点60(add_hair/0716add-hair-api/hair_repaint) + 全部测试页输入框
 2026-07-27 23:36:15 +08:00
xsl b354c8c627 fix: 删除 MeasureResult.__init__ 中重复的七眼厘米赋值块 2026-07-27 23:17:35 +08:00
xsl fd9c41430c feat: 接口1/5/6 发际线弃用逻辑(顶庭<0.7cm) 
发际线离头顶<0.7cm时判定分割不可靠，弃用发际线：
顶/上庭字段置null、face_total只算中下庭、标注图保留头顶线去掉发际线、
只标中/下庭。eye1/7竖向范围改用眉心。
 2026-07-27 23:12:33 +08:00
xsl a1d458eb20 feat(接口1/5/6): 返回数据新增 left_position/right_position(MediaPipe 21/251号点) 
- face_mesh_landmarks.py: 加常量 LEFT_POSITION=21 / RIGHT_POSITION=251
- measure.py: MeasureResult 收 landmarks/宽高, to_response 顶层输出两点(原图像素 {x,y}, 与 landmarks 同格式)
- measure_face 透传 landmarks(签名不变, 6处调用零改动); __init__ 用 None 默认值守卫向后兼容
- 三接口自动生效: 接口1/6 在 data 顶层, 接口5 在 face_measure 对象里(复用同一 to_response)
- 实测坐标左右镜像合理, 44 个现有测试全过无回归
 2026-07-24 00:42:26 +08:00
xslandCursor e7b62f2b2e perf(接口2): 稳定混跑耗时至12s内 —— ComfyUI插队 + CLIP挪CPU + 提示词全局统一 
问题：接口2 与接口3/5 乱序调用时耗时抖动（最差 15~22s）。两个根因：
1. GPU 24G 常驻 21.4G，Flux-2(3.9G) 无法完全驻留显存，每次采样动态换页，
   速度随空闲显存波动（2s~8s）；
2. ComfyUI 单队列 FIFO，接口2 排在接口3/5 批量任务后面。

改动：
- hairline/comfyui.py: run() 新增 front 参数，/prompt 带 "front": true 插队到队列最前；
  redraw.py 透传；service.py 接口2 三处调用（女重绘 + 男有/无遮罩）传 front=True，
  接口3/5 仍走普通队列。
- add_hair.json / 0716add-hair-api.json: 节点61 CLIPLoader device default→cpu。
  qwen CLIP(4G) 不再占显存（文本条件缓存常年命中），ComfyUI 显存 8.8G→4.5G，
  Flux-2 完全驻留，采样稳定 ~3-5s。代价：换 prompt 后首次请求 CPU 编码 ~11s（一次性）。
- 提示词全局统一为「填充遮罩区域的头发，皮肤加一点磨皮，再加一点美颜」：
  app.py 4处默认值、service.py _REDRAW_PROMPT、redraw.py _DEFAULT_PROMPT、
  4个工作流节点60内置文案、测试页(test_interface2/3/7/12/12_final)、local_test。
  任何两个不同 prompt 交替提交都会打爆 CLIP 编码缓存（--cache-classic 只存最近一次），
  之前测试页旧文案与服务端不一致导致交替测试每次 +11s。
- app.py: 接口7 /api/v1/hair/grow-v2 下线（业务弃用；add_hair2.json 的 Klein-9b
  会把常驻 Klein-4b 挤出显存）。保留 stub 返回 1007 明确报错，避免裸 404。

实测（1024 档）：接口2女 8.5~10s、接口2男 ~5s、接口3 ~7-10s，交替混跑无尖刺。

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-23 01:01:48 +08:00
xslandCursor 7fc0210ce6 fix(pose): 修正正面照被误判为1003（solvePnP翻转解） 
ITERATIVE 偶发收敛到相机后方(tz<0)，roll≈±180° 超阈值，
把正面照误判为非正面。检测到负深度时回退 SQPNP 重解正深度解。
补充回归测试。

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-21 16:47:04 +08:00
xslandCursor 085ad3ced0 接口1 四庭七眼标注：字体更小、数值带cm、增加百分比、线名右移居中对齐 
Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-19 23:37:59 +08:00
xsl 74ccab0ff8 feat: 适配 RTX 3090 (24GB) 环境优化 
硬件迁移：从 RTX 5090 (32GB) 迁移到 RTX 3090 (24GB)

主要改动：
1. hairline/comfyui.py: 轮询间隔从 0.2s 降到 0.05s
2. hairline/service.py: PNG 编码 compress_level=1，节省 ~240ms
3. add_hair.json: 工作流使用 4B FP8 模型 + steps=4
4. static/test_interface3.html: 修复图片显示（添加 data:image/jpeg;base64, 前缀）

性能优化后接口3响应时间：6.6-7.3s（之前 8.66s）
 2026-07-18 18:58:40 +08:00
xslandCursor 659c037270 feat: ComfyUI 改走 10.60.74.221，测试页上传图超阈值自动降采样 
将 worker 默认 ComfyUI 地址改为远端 10.60.74.221:8188；前端测试页在像素超过 1536000 时等比缩小到 786432 以内。

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-17 02:23:35 +08:00
xsl 2b1f528ddd perf: 接口2/12 跳过 viz 叠图生成，省 ~80ms 
compute_mask 新增 render_viz 开关（默认 True 保持接口11 行为不变）。
接口2/12 路径(generate_hairline_redraw)传 render_viz=False，跳过 6+ 张
overlay JPG 的 _jpg_b64 编码（baseline/upper/hair_seg/mask/hairline/pushed），
只保留必需的数据字段（_inner_pts/_outer_pts/_upper_mask/mask_pixels）。
 2026-07-17 01:37:53 +08:00
xsl 5fbc03a6df feat: 接口12/接口2 发际线重绘改走 local_test 外部 ComfyUI 服务 
后端 generate_hairline_redraw 跳过内置 Flux-2 重绘，改为产出 final(接缝融合基底)
+ 纯红遮罩 PNG(redraw_band_mask_base64，遮罩区=(255,0,0,255)、其余全透明)。

接口2 female(generate_grow_results_swap) 取 final+遮罩后在后端调 local_test
(0716add-hair.json 工作流) 完成重绘，结果作为生发图返回；male 分支不变。

测试页 test_interface12.html / test_interface12_final.html 改为两阶段：
先生成 final+纯红遮罩，再前端调 local_test 重绘并展示；color_match 默认不勾选。

local_test/app.py 加 CORS 头(OPTIONS 预检)，支持浏览器跨域直连。
 2026-07-17 00:00:15 +08:00
xslandCursor e8a2c5a8a1 fix: 日志目录去掉硬编码 /home/xsl，改为基于仓库根解析（支持 HAIR_LOG_DIR 覆盖） 
worker 部署到 /home/ubuntu 等其他路径时，原硬编码 /home/xsl/hair/log
会导致 Permission denied。改为相对仓库根，兼容多机部署。

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-16 08:07:03 +08:00
xslandCursor a208fe88ec feat: 拆分接口11/12，新增接口12 final / final v2 精简重绘端点 
- 接口11 移除重绘，仅生成 final；接口12 (grow_v2) 负责发际线带 Flux-2 重绘
- 接口12 重绘带改为发际线外推 band_lo_mult~band_hi_mult 倍 push（默认 0.5~1.5），页面可调
- 接口12 同时产出 A 整帧重绘 与 B 局部加发+全脸美颜（beauty_alpha 可调）
- 新增 grow_v2_final（整帧重绘）/ grow_v2_final_v2（B 局部+美颜）端点：仅需图片+发型 ID，其余用固化默认值（color_match 关闭）
- 配套精简测试页 test_interface12_final.html / test_interface12_final_v2.html / test_interface12.html
- 恢复接口11 调试页多频段与换发型可调参数、color_match 默认不勾选
- 删除旧脚本 batch_grow_v2.py / gen_report_hairline_v2.py / test_simple.html

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-16 00:04:36 +08:00
xsl 4001df2c34 save code 2026-07-15 01:16:32 +08:00
xsl b163a3f34a 调整 baseline 关键点 + 新增极简测试页 
- head_mask.py: BASELINE_IDX 改为 [162,71,68,104,69,108,151,337,299,333,298,301,389]，
  左端点 21→162、右端点 251→389，新增 71/301 两点；同步更新注释
- static/test_simple.html: 极简测试页，仅需选图片+发型，原图与结果左右并排对比
- .gitignore: 忽略 report_hairline_v2.zip
 2026-07-12 22:27:02 +08:00
xsl 1e6c2e54de 接口11/12：固定 pushed 遮罩 + multiband 融合，移除其他算法选项 
遮罩算法只保留 pushed（发际线外推），融合算法只保留 multiband（多频段金字塔），
eroded/closed/feather/alpha_gradient/seamless 等旧选项从接口参数层移除。

- generate_hairline_grow: 删除 mask_type/blend_method/feather_px/color_match 参数，
  内部固定 mask_type=pushed、blend_method=multiband
- app.py 接口11/12: 删除 mask_type/blend_method/feather_px/color_match Form 参数，
  调用处改关键字传参；grow_v2 只传 image+hairline_id 即默认走最新算法
- 前端两个测试页: 删除遮罩/融合下拉选项及相关联动，固定展示 pushed 步骤
- 文档: 更新为"固定 pushed + multiband，移除其他选项"
 2026-07-11 23:31:19 +08:00
xsl 1b9f3fdb6f 终于修改对了，还是opus 2026-07-11 23:14:07 +08:00
xsl 41bb164a52 接口11/12：新增发际线外推遮罩(pushed)模式 + multiband融合修复 + 调试日志 + 对比报告脚本 
发际线生发遮罩算法（mask_type=pushed）：
- _extract_hairline：提取头发/皮肤交界线（逐列头发下沿），用 baseline 水平 y 线截断（无竖线）
- _pushed_mask：以眉心(151点)为圆心逐点径向外推 push_cm，与 baseline 组闭合区域
- 径向归并锯齿用插值填补，避免遮罩碎裂
- pushed 模式过程可视化（①-f 交界线 / ①-g 外推+遮罩），eroded/closed 不展示无关步骤

multiband 金字塔融合修复（hairline_grow.py）：
- mb_levels 按层数膨胀外缘 keep 区，让过渡带随层数变宽（旧硬二值钳回导致 mb_levels 形同虚设）

接口12 grow_v2（固定参数精简版）：
- 固定 multiband/mb_levels=5/erode_cm=0.6，仅返回 final_base64
- 支持 mask_type=pushed + hairline_push_cm/hairline_edge

调试支持：
- 调试页 test_interface11_debug.html（前后端日志面板 + 下载日志按钮）
- hairline_grow.log 全链路日志（按 rid 关联），/api/v1/debug/hairline_log 下载接口
- 遮罩计算过程可视化（baseline/upper/头发分割/交界线/外推/最终遮罩）

文档与脚本：
- docs/发际线生发遮罩算法_pushed模式.md 算法说明
- scripts/batch_grow_v2.py 批量调用、gen_report_hairline_v2.py 对比报告生成
 2026-07-11 22:23:57 +08:00
xslandClaude 864d7f969a fix: torch.load 添加 weights_only=False 以兼容新版 PyTorch 
Co-Authored-By: Claude <noreply@anthropic.com>
 2026-07-09 23:44:48 +08:00
xslandClaude Opus 4.8 0ddfa83743 接口11：发际线生发（接口9遮罩 + change_hair换发型/区域生发 + 按遮罩羽化贴回）+ 分步可视化测试页 
- 端点 POST /api/v1/hairline/grow（app.py，纯新增，不影响接口1-5）
- 编排模块 face_analysis/hairline_grow.py：复用接口9 遮罩 → HTTP 调 change_hair(8801) → 按遮罩贴回
- 双生成后端 gen_backend：swaphair(换发型LoRA) / hairgrow(区域生发inpaint)
- swap_mode：ext_mask(接口9遮罩作换发型遮罩) / as_is；融合 feather/alpha_gradient/seamless
- 参数全在测试页可调；可视化按算法文档4步：最终遮罩→生成全帧→严格贴回→接缝融合
- 附启停脚本 scripts/restart_if11_backends.sh、算法文档、测试图

注：change_hair 侧 swapHair 的 ext_mask/denoising_strength 改造在 change_hair 仓库，向后兼容。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-07-09 22:06:17 +08:00
xslandClaude Opus 4.8 3eb60bddc5 接口10：头部外缘膨胀带遮罩 + 分步可视化 
新增 POST /api/v1/head/band（worker + 网关代理）与测试页 test_interface10.html：
- 先和接口9 一样得到内缩后的基准遮罩（含额头的闭合区域外缘朝151内缩 erode_cm、
  底线不动，默认1.2cm），在此基础上取外轮廓线，去掉贴着底部分割线的那一段。
- 把外轮廓线膨胀成带子（总宽 dilate_cm，默认2cm；半径=总宽/2，虹膜标定换算）。
- 裁到分割线以上（不越过底线）。BiSeNet/SegFormer 两套并排对比。
- 两个可调参数 erode_cm + dilate_cm（页面数字框+滑块+localStorage）。
- 复用 head_mask 的构件，无新依赖；纯新增，不改动既有接口。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-07-07 23:32:45 +08:00
xslandClaude Opus 4.8 9774997035 接口9：头发遮罩生成 + 分步可视化 
新增 POST /api/v1/head/mask（worker + 网关代理）与测试页 test_interface9.html：
- MediaPipe 关键点连成额头分割线（21,68,104,69,108,151,337,299,333,298,251，
  左端21/右端251 水平延伸到图片边缘），分割线以上为上半区。
- 头发分割 BiSeNet 与 SegFormer 两套并排对比；每列从最顶端头发向下填充到分割线，
  得到含额头的闭合区域（不从发际线割断）。
- 外缘朝中心点151内缩 erode_cm（默认1.2cm，页面可调，虹膜标定换算像素）、底线不动。
- 复用现有 detector/hair_segmenter/SegFormer 单例（只读推理），无新依赖；纯新增，
  不改动既有接口。

顺带修复接口2 遗留测试 test_grow_female_returns_5：hair_style 自 cb1989c 起必填，
补上 hair_style=1,2,3,4,5。全套 42 passed。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-07-07 23:09:29 +08:00
xslandClaude Opus 4.8 a9439e4975 接口1: 标注图人头最左/最右竖线改用耳朵分割外缘 
- 最左/最右竖线由「头发轮廓」改为同一 BiSeNet 的耳朵类(7/8)外缘，
  看不到耳朵(被头发/侧脸遮挡→掩膜空)则该侧不画线
- 外耳轮廓常被误标成头发: 从耳朵外缘沿紧邻前景(耳∪发)按脸宽自适应
  外扩回收(上限 face_w*0.045, 遇背景间隙即停)
- 先按人脸包围盒裁剪再分割: BiSeNet 在紧裁人脸上训练, 整张全身/街拍图
  脸偏小会严重欠分割、丢耳朵; 裁剪后映射回原图, 耳朵稳定可分
- segment_hair_and_ears() 单次推理同出 hair_mask + ear_mask

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-24 21:36:23 +08:00
xsl 57d1a876cc save code 2026-06-23 23:26:40 +08:00
xsl 6e8113b3ff save code 2026-06-23 22:46:24 +08:00
xsl 85f1ca521d 修改画线 2026-06-23 22:29: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
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
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 7493f48992 加入模型文件 2026-06-14 00:24:09 +08:00