98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
"""穷举 process_image 的参数组合，找出哪种组合会产出"未紧致裁剪"的输出。
|
||
|
||
逻辑:
|
||
final.png (153x74, bbox=(0,0,153,74)) 是已知正确的紧致输出。
|
||
遍历 is_signature/height_cm 等参数，跑 process_image，对比输出:
|
||
- 尺寸 == (153, 74)?
|
||
- bbox 是否贴满？
|
||
- 字节内容是否与 final.png 完全一致？
|
||
任何一项不一致 -> 找到 bug 触发条件。
|
||
|
||
输入图: image_test/input/test1.jpg
|
||
"""
|
||
import io
|
||
import sys
|
||
import hashlib
|
||
from pathlib import Path
|
||
from PIL import Image, ImageChops
|
||
|
||
BASE_DIR = Path(__file__).resolve().parent
|
||
PROJECT_ROOT = BASE_DIR.parent
|
||
INPUT_IMAGE = BASE_DIR / "input" / "test1.jpg"
|
||
FINAL_REF = BASE_DIR / "output" / "test1" / "final.png"
|
||
|
||
|
||
def _ensure_path():
|
||
if str(PROJECT_ROOT) not in sys.path:
|
||
sys.path.insert(0, str(PROJECT_ROOT))
|
||
|
||
|
||
def _stats(data):
|
||
img = Image.open(io.BytesIO(data))
|
||
if img.mode != 'RGBA':
|
||
img = img.convert('RGBA')
|
||
r, g, b, a = img.split()
|
||
luma = Image.merge('RGB', (r, g, b)).convert('L')
|
||
am = a.point(lambda p: 255 if p > 30 else 0)
|
||
lm = luma.point(lambda p: 255 if p < 210 else 0)
|
||
combined = ImageChops.darker(am, lm)
|
||
total = img.size[0] * img.size[1] or 1
|
||
return {
|
||
'size': img.size,
|
||
'bbox': combined.getbbox(),
|
||
'content_pct': 100.0 * sum(combined.histogram()[1:]) / total,
|
||
'sha': hashlib.sha256(data).hexdigest()[:12],
|
||
'bytes': len(data),
|
||
}
|
||
|
||
|
||
def main():
|
||
_ensure_path()
|
||
sys.stdout.reconfigure(encoding='utf-8')
|
||
|
||
if not INPUT_IMAGE.exists():
|
||
print(f"缺少: {INPUT_IMAGE}")
|
||
sys.exit(1)
|
||
if not FINAL_REF.exists():
|
||
print(f"缺少: {FINAL_REF}，请先运行 image_test/process.py 生成")
|
||
sys.exit(1)
|
||
|
||
with open(INPUT_IMAGE, 'rb') as f:
|
||
image_bytes = f.read()
|
||
with open(FINAL_REF, 'rb') as f:
|
||
final_bytes = f.read()
|
||
|
||
final_stats = _stats(final_bytes)
|
||
print(f"=== 参考基准 final.png ===")
|
||
print(f" 尺寸={final_stats['size']} bbox={final_stats['bbox']} "
|
||
f"内容={final_stats['content_pct']:.1f}% 字节={final_stats['bytes']} "
|
||
f"sha={final_stats['sha']}")
|
||
print()
|
||
|
||
from lib.image_processor import process_image
|
||
|
||
cases = [
|
||
# (label, kwargs)
|
||
("is_sig=True noise=0 rot=0", {'is_signature': True}),
|
||
("is_sig=False noise=0 rot=0", {'is_signature': False}),
|
||
("is_sig=True noise=3 rot=0", {'is_signature': True, 'noise_level': 3}),
|
||
("is_sig=True noise=0 rot=-5,5",{'is_signature': True, 'rotate_min': -5, 'rotate_max': 5}),
|
||
("is_sig=True noise=3 rot=-5,5",{'is_signature': True, 'noise_level': 3, 'rotate_min': -5, 'rotate_max': 5}),
|
||
]
|
||
|
||
print(f"=== 穷举 {len(cases)} 种参数组合 ===\n")
|
||
for label, kw in cases:
|
||
out = process_image(image_bytes, log_source='测试', **kw)
|
||
s = _stats(out)
|
||
tight = (s['bbox'] == (0, 0, s['size'][0], s['size'][1]))
|
||
same_as_final = (out == final_bytes)
|
||
flag = "✓紧致" if tight else "✗有padding"
|
||
flag2 = "=final" if same_as_final else "≠final"
|
||
print(f"[{label}]")
|
||
print(f" 尺寸={s['size']} bbox={s['bbox']} 内容={s['content_pct']:.1f}% "
|
||
f"字节={s['bytes']} {flag} {flag2}")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|