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
13 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
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
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 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 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