xsl/hair
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
98 Commits 4 Branches 0 Tags
c1bb9614c78fc87af7136e817d9d24dd14d14443
Commit Graph
98 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
xsl c1bb9614c7 save code 2026-07-17 01:54:04 +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 d9e96aca87 chore: 重绘服务改走内网 10.60.74.221:8899 
gpu_worker 与重绘服务在同一内网，走内网地址（3ms 延迟，比公网更稳定）。
三处调用地址（service.py / test_interface12 / test_interface12_final）
由公网 117.50.183.232 改为内网 10.60.74.221。仍可用 HAIR_LOCAL_REDRAW_URL 覆盖。
 2026-07-17 00:57:58 +08:00
xsl 8aed389d79 chore: 重绘服务拆到远程机器 117.50.183.232:8899 
接口2 female 后端(generate_grow_results_swap)、test_interface12.html、
test_interface12_final.html 三处重绘调用地址由本机 127.0.0.1:8899 改为远程
117.50.183.232:8899，本机不再跑重绘服务（释放本机显存）。

地址仍可用 HAIR_LOCAL_REDRAW_URL 环境变量覆盖。
 2026-07-17 00:44:20 +08:00
xsl c72e3ceda9 asdf 2026-07-17 00:32:13 +08:00
xsl 99ce21334a asdf 2026-07-17 00:26:47 +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
xsl 0bbb15d668 添加服务 2026-07-16 22:57:12 +08:00
xsl 632e75317b 接口6 增加字段 2026-07-16 12:38:10 +08:00
xsl 1cd4115b26 Merge branch 'main' of http://git.xiangsilian.com:3000/xsl/hair 2026-07-16 09:38:53 +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
xsl 0f8a7e27c0 部署线上5090时候的改动git commit -m 部署线上5090时候的改动。 2026-07-16 07:54:07 +08:00
xsl 12f34c44f2 修改发型为最终发型 2026-07-16 00:22:54 +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 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 d5397a6d6f chore: 接口2/3/5/7 生发 ComfyUI 默认提示词加「加一点美颜」 
将接口2（/api/v1/hair/grow）、接口3（/api/v1/hair/grow-b）、
接口5（/api/v1/hairline/generate）、接口7（/api/v1/hair/grow-v2）的
prompt 参数默认值从「补充遮罩区域的头发」改为「补充遮罩区域的头发，加一点美颜」。
 2026-07-13 23:41:42 +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
xsl 28255ef7c2 feat: 接口5 新增 face_measure（复用接口1测量数值）+ 接口1 七眼 eye1~eye7 
接口5（/api/v1/hairline/generate）在发际线结果基础上新增 data.face_measure，
复用接口1的四庭七眼测量数值（四庭/七眼 eye1~eye7/landmarks/姿态），
不含标注图；独立流程容错，测量失败时为 null 不影响发际线主结果。

重构：提取 _run_face_measure_data() 共用函数，接口1/6 改调它再补标注图，
行为不变（43 测试全过，错误码 1001/1003/1008 回归正常）。

接口1/5 七眼新增 eye1~eye7 从左到右 7 段宽度（cm），eye1/eye7 耳朵不可见
时为 null（保留键）。

测试页 test_interface5.html：移除点击切换卡片网格改为所有发型平铺，
新增四庭七眼测量卡片。前端接入页 integration.html / 接口文档同步更新。
 2026-07-13 22:40:55 +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
Ubuntu 5d9d91bc82 save code 2026-07-12 18:50:05 +08:00
xsl 3fe5f6cf0d fix: batch_grow_v2.py 的 hr_options 从 HR_OPTIONS 动态生成，不再硬编码两档 
旧代码硬编码 hr_options=[hr,nohr]，导致仅非高清批量跑完后 meta 仍含 hr 档，
报告脚本取 hr_opts[0]=hr 与实际结果的 nohr 不匹配，结果图全部渲染不出来。
 2026-07-12 00:08:01 +08:00
xsl e17677158a 批量测试：仅非高清（20脸×5发型=100张）+ 报告原图大图并排对比 
- batch_grow_v2.py: HR_OPTIONS 仅保留非高清档，去掉高清
- gen_report_hairline_v2.py: 每张脸 grid 改为「原图大图 + 5发型」6列并排，
  原图不再是缩略图；去掉高清/非高清行标签和失败表高清列
 2026-07-12 00:02:42 +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
Ubuntu de2cecf0d5 save code 2026-07-10 00:38:39 +08:00
xslandClaude Opus 4.8 f8e30ad32e 接口5：改为多选发型 + 每发型返回 middle/high/low 三档叠图（去掉 hairline_level 入参） 
- 去掉 hairline_level 入参；middle/high/low 三档都返回
- 入参改为同接口2：gender + hair_style(逗号分隔多选，必填)，缺失/越界返回 1007
- 每个选中发型返回 image_{middle,high,low}_base64 三档叠图 + grown_image_base64 生发图，
  按发型分组；order = 发型序号，含 hairline_type
- 生发黑模板仍固定 middle(hairline_texture_black/)，每发型 1 张生发图
- best_hairline_center_point 取首个选中发型的 middle 档
- 同步更新测试页(三档并排展示)、integration.html、stub_worker、接口文档/实现说明、test_api

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-07-10 00:19:08 +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 9675e147a0 接口5：新增 hairline_level 档位(middle/high/low)贴图 + 集成接口2生发能力 
- hairline_level: 可选 middle(默认)/high/low，选用不同高度档位发际线贴图；
  新增 hairline_texture_high/low 两套贴图，get_texture_map 改为按档位缓存
- hair_style: 可选逗号分隔序号，对选中发际线类型同步生发(ComfyUI)，
  结果合并进 hairline_images[].grown_image_base64(未选中为 null)；
  生发黑模板固定取 middle(hairline_texture_black/)，与档位无关
- 新增 use_mask/prompt 生发控制参数(同接口2)
- 测试页/接口文档同步更新

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-07-09 23:30:21 +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
Ubuntu aa981229c0 添加统计请求的功能 2026-07-02 23:28:49 +08:00
Ubuntu 5d94edfc1a save code 2026-06-25 20:25:49 +08:00
xslandClaude Opus 4.8 a177dc2583 去除上传图片的分辨率与文件大小限制 
- app.py: 删除全部 MAX_FILE_BYTES(≤1MB→1006) 与 MIN_SHORT_SIDE/MIN_LONG_SIDE
  (分辨率→1002) 校验及对应常量; 同步清理 File 描述、图片要求说明、
  错误码表(移除1002/1006)与过时示例
- gateway/app.py: 删除注释掉的 1006 大小校验块与描述里的 ≤1MB
- run_worker.sh / hair-worker.service: 删除临时放开限制的环境变量
- tests: 移除已过时的 test_oversize_1006 / test_lowres_1002 及 oversize_file fixture

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
 2026-06-24 22:21:39 +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
Ubuntu 5fd7d05d26 save code 2026-06-24 20:49:34 +08:00
Ubuntu e58ece0ada save code 2026-06-23 23:30:24 +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 8cd44848d6 feat(接口6): 复刻接口1，新增 /api/v1/face/measure-v2 
- 抽取 _face_measure_impl() 共用实现，接口1/6 零逻辑差异
- 接口6 路径 POST /api/v1/face/measure-v2
- 入参/出参与接口1 完全一致
- 文档同步更新（接口文档、实现说明、网关待改动）

Co-Authored-By: Claude <noreply@anthropic.com>
 2026-06-23 20:27:51 +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
xslandClaude 9678b54267 feat(接口2/7): hair_style 支持逗号分隔多选，如 1,2,3 
- 参数类型从 int 改为 string（逗号分隔），自动解析去重排序
- 越界/非法值返回 1007
- _parse_hair_styles() 辅助函数：解析 + 去重 + 范围校验
- service 层 hair_style 参数改为 hair_styles: list[int]
- 接口2 和 接口7 同步更新
- 服务已重启，smoke test 通过

Co-Authored-By: Claude <noreply@anthropic.com>
 2026-06-22 23:36:13 +08:00
xsl 1d23414d21 save code 2026-06-22 23:23:46 +08:00
xslandClaude cb1989c042 feat(接口2/7): 接口2加hair_style参数选单张发型；新增接口7用add_hair2工作流 
接口2 变更：
- 新增必填 hair_style(int) 参数，按序号只生成一张(不再全量)
- female:1-5 male:1-4，越界返回1007

接口7 新增：
- POST /api/v1/hair/grow-v2，功能与接口2一致
- 使用 add_hair2.json 工作流(Flux-2 Klein 9b)
- SaveImage输出节点自动检测(75)

comfyui.py 重构：
- run() 支持 workflow_path 参数，多工作流按路径缓存
- SaveImage 输出节点自动检测，不再硬编码
- 输入/种子/提示词节点ID两个工作流相同(26/6/60)

文档：
- 接口文档、实现说明、网关待改动 三份同步更新
- 网关只需加一行路由，base64→URL改写无需改动

Co-Authored-By: Claude <noreply@anthropic.com>
 2026-06-22 23:22:25 +08:00
xslandClaude Sonnet 4.6 28a6062fea feat(接口4): 精简输出为固定6个英文字段，缩短提示词 
- face_features.py：_PROMPT 只问6项特征（+有无人脸），analyze_features
  只返回6个英文字段；无人脸返回 None（不再依赖 has_face 在外部判定）
- gateway/app.py：无脸判定改为 feats is None，去掉 has_face import
- app.py / docs / test_interface4.html：Swagger/文档/示例/测试页同步

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
 2026-06-18 22:19:00 +08:00
xsl be81567261 save code 2026-06-17 23:36:01 +08:00
Ubuntu 24da4eae6f save code 2026-06-17 23:27:34 +08:00