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
16 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 5101bb5f6b feat(接口2): 测试页暴露重绘分辨率选项 + 分辨率对比测试套件 
- test_interface2.html: 新增「重绘分辨率」下拉(默认896/1024/768/640/不缩图0)，
  仅 female 生效，append redraw_max_side 字段；male 时自动隐藏
- 新增分辨率对比测试工具 (image/compare_test, image/res_test):
  - batch_test.py: 串行批量测试脚本，支持断点续跑
  - gen_report.py: 生成自包含 HTML 对比报告(速度色阶+缩略图+点击放大)
- 两轮测试结果:
  - v1: 4图×5发型×2档=40次 (compare_test)
  - v2: 5图×3发型×5档=75次 (res_test)
  - 结论: 大图(长边>1600)原图直送比896档慢~3倍; 中小图各档差异小
- .gitignore: 忽略测试结果图/原图副本/运行日志(out/, *.log, progress.json)
 2026-07-26 22:06:27 +08:00
xsl ff4019c570 chore: 默认模型固定为 9B-Q4_K_M，测试页移除模型/分辨率选择器 
测试结论: 9B-Q4_K_M 分辨率896 为最佳性价比组合。
- 三个工作流(0716add-hair-api/add_hair/hair_repaint)默认模型改为
  flux-2-klein-9b-Q4_K_M.gguf (UnetLoaderGGUF节点)
- test_interface2.html 移除 Flux模型/压图长边 下拉选择器，
  接口调用不再传 flux_model/redraw_max_side，统一用工作流默认(Q4/896)
 2026-07-25 16:25:44 +08:00
xsl 94cdfd6de5 feat(接口2): 支持动态切换Flux模型+分辨率 + 模型对比测试脚本 
代码改动:
- comfyui.py: run() 新增 unet_name 参数，提交前自动改写模型节点
  (.gguf→UnetLoaderGGUF, .safetensors→UNETLoader)，并按模型自动同步
  文本编码器(4b→qwen_3_4b, 9b→qwen_3_8b)，避免切换时维度不匹配
- redraw.py: run_redraw() 透传 unet_name
- service.py: generate_grow_results_swap/generate_grow_results 支持
  redraw_max_side(分辨率参数化) 和 unet_name 透传
- app.py: 接口2 新增 flux_model/redraw_max_side 两个 Form 参数(男女路径都加)
- test_interface2.html: 新增 Flux模型/压图长边 下拉选择器
- add_hair.json/0716add-hair-api.json: 工作流默认模型改为 9b

测试脚本:
- benchmark_matrix.py: 4模型×3分辨率×3图×3次 矩阵测试
- benchmark_hairstyle.py: 3图×5发型×10组合 发型对比测试
- benchmark_report.py/benchmark_hairstyle_report.py: HTML报告生成

清理:
- .gitignore: 排除 benchmark_out/、报告HTML、gateway.log、*.bak.*
- 移除 gateway.log 的 git 跟踪
 2026-07-25 16:16:13 +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 08b31a3baa fix: 测试页同时兼容 base64 与 URL 图片字段 
直连 worker 返回 *_base64、经网关则改写为 *_url；统一 resolveImgSrc 后接口1/2/3/5/6 测试页都能正确显示结果图。

Co-authored-by: Cursor <cursoragent@cursor.com>
 2026-07-19 15:49:46 +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 12f34c44f2 修改发型为最终发型 2026-07-16 00:22:54 +08:00
xsl 9386f84c88 docs: 同步默认提示词「加一点美颜」到文档/接入页/测试页 
- docs/接口文档.md：接口2/3/5/7 参数表统一更新默认提示词，接口2/3/7
  补全此前缺失的 prompt 参数行；顺带修正接口7 输出表 image_url 描述
  为透明 PNG（上一轮透明 PNG 改动的遗漏）
- static/integration.html：接口2 描述句 + 接口7 字段表同步透明 PNG 描述
- static/test_interface2/3/7.html：prompt 输入框默认值同步更新
 2026-07-13 23:46:40 +08:00
xsl 95c6a2d929 feat: 接口2/5 发际线叠图改为透明 PNG（仅曲线），生发图不变 
接口2（/api/v1/hair/grow）的 image_url 和接口5（/api/v1/hairline/generate）
的 image_middle/high/low_url 从「原图+白线合成 JPG」改为「透明底 PNG（仅含
发际线曲线）」，前端需叠加原图显示。grown_image_url 生发图保持不变（ComfyUI
完整人像照片）。

实现：
- app.py 新增 _rgba_png_b64() 编码 RGBA 透明层为 PNG base64（保留 alpha）
- hairline/service.py 接口2/5 改用 build_overlay_layer（返回 RGBA 透明层）
  替代 render_hairline_overlay（合成到原图）
- 网关零改动：rewrite_base64_to_url 已按 \x89PNG 魔数嗅探落盘为 .png

测试页：test_interface2/5.html 改为「原图打底 + 透明PNG 绝对定位叠加」显示
（复用 .img-stack 结构，固定叠加无开关）。

文档：接口文档.md / integration.html 更新接口2/5 图片字段说明。
测试：43 passed，接口2 image_base64 改断言为 PNG 魔数，接口5 三档叠图同。
 2026-07-13 23:38:55 +08:00
Ubuntu aa981229c0 添加统计请求的功能 2026-07-02 23:28:49 +08:00
Ubuntu 5d94edfc1a save code 2026-06-25 20:25:49 +08:00
UbuntuandClaude 76bae98073 feat(接口2/7): hair_style改为多选复选框，新增接口7测试页，网关注册v2路由 
- 接口2/7测试页：发型从单选下拉框改为多选复选框+全选/全不选，提交时逗号分隔
- 新建static/test_interface7.html（基于接口2测试页，紫色主题，调用/api/v1/hair/grow-v2）
- gateway/app.py：注册/api/v1/hair/grow-v2代理路由 + 首页索引
- integration.html：新增接口7完整文档卡片（入参/出参/测试页链接）

Co-Authored-By: Claude <noreply@anthropic.com>
 2026-06-22 23:43:39 +08:00
Ubuntu 24da4eae6f save code 2026-06-17 23:27:34 +08:00
Ubuntu c15a3c2e15 save code 2026-06-17 23:04:49 +08:00
Ubuntu cd4ca278d4 完成接口1 和接口2，测试页面也完成 2026-06-15 10:10:52 +08:00