#!/usr/bin/env python3 """生成 v3 wave发型分辨率对比报告：每行=原图+5档，21行；含各档耗时统计。""" import json, os, html OUT = "/home/ubuntu/hair/image/wave_test/out" REPORT = os.path.join(OUT, "report.html") results = json.load(open(os.path.join(OUT, "report.json"))) SIDES = [("origin", "原图直送", 0), ("s1024", "1024", 1024), ("s896", "896", 896), ("s768", "768", 768), ("s640", "640", 640)] # 图片顺序（与测试脚本一致）：girl_img 排序 + asdf + qwer IMG_DIR = "/home/ubuntu/hair/image" IMAGES = [] for f in sorted(os.listdir(os.path.join(IMG_DIR, "girl_img"))): if f.lower().endswith((".jpg", ".jpeg", ".png")): IMAGES.append(os.path.splitext(f)[0]) IMAGES.append("asdf") IMAGES.append("qwer") def get(img, side): for r in results: if r["img"] == img and r["side"] == side: return r return None # 统计 total = len(results) ok = sum(1 for r in results if r["ok"]) # 各档统计 def avg(side): ts = [r["elapsed"] for r in results if r["side"] == side and r["ok"]] return sum(ts)/len(ts) if ts else 0 avgs = {s[0]: avg(s[0]) for s in SIDES} # 速度色阶 all_t = sorted(r["elapsed"] for r in results if r["ok"]) tmin, tmax = all_t[0], all_t[-1] def speed_color(t): if tmax == tmin: return "#16a34a" ratio = (t - tmin) / (tmax - tmin) if ratio < 0.33: return "#16a34a" elif ratio < 0.66: return "#f59e0b" else: return "#dc2626" # 表格行 rows_html = [] for img in IMAGES: orig_cell = (f'' f'

' f'

📷 原图

') side_cells = [] for side_name, side_lbl, side_val in SIDES: r = get(img, side_name) if r and r["ok"]: col = speed_color(r["elapsed"]) cell = (f'' f'

' f'

{r["elapsed"]}s

') elif r: cell = (f'

❌

' f'

{html.escape(r["err"][:30])}

') else: cell = '

—

' side_cells.append(cell) rows_html.append("" + orig_cell + "".join(side_cells) + "") # 各档统计卡 def stat_card(lbl, val, col, rng=""): return (f'

' f'

{val:.1f}s

' f'

{lbl}{rng}

') stat_cards = "".join([ stat_card("原图直送(0)", avgs["origin"], "#dc2626"), stat_card("1024", avgs["s1024"], "#f59e0b"), stat_card("896", avgs["s896"], "#2563eb"), stat_card("768", avgs["s768"], "#16a34a"), stat_card("640", avgs["s640"], "#0d9488"), ]) html_doc = f""" wave发型 5档分辨率对比

💇 wave发型 · 5档分辨率对比报告

POST /api/v1/hair/grow · female · wave(波浪) · 21 图 × 5 档分辨率 = 105 次 · 串行 · 4090 (24G)

{ok}/{total}

成功 / 总数

{stat_cards}

📊 结论：耗时随分辨率单调下降。大图(asdf 1666/qwer 1678) 原图直送需 24~26s，是 1024 档(11s) 的 2.3 倍；中小图(≤1024) 各档差异较小(6~13s)，因小图本身不触发缩图。 4090 24G 全程无 OOM，105/105 成功。画质对比见下表（横向滑动），点击任意图可放大。快(<{(tmin+(tmax-tmin)*0.33):.0f}s) 中等慢(>{(tmin+(tmax-tmin)*0.66):.0f}s)

{"".join(rows_html)}

📷 原图	原图直送 (0) 不缩放	1024	896 默认	768	640

""" with open(REPORT, "w") as f: f.write(html_doc) print(f"已生成: {REPORT}")