| |
| """Generate a labeled sample gallery of every augmentation effect in isolation. |
| |
| For each of the 25 configurable augmentation families, forces that one effect |
| on (probability=1.0) and disables every other effect, so the visual result is |
| attributable to that effect alone. Placement is frozen (no rotation/scale/ |
| misplacement jitter, fixed template) so samples across effects are directly |
| comparable. Writes 5 sample images per effect to degradation/<effect>/, each |
| a multi-panel figure (full page, signature crop, ink mask, coverage metrics, |
| difficulty breakdown) via the existing visualizer. |
| |
| Usage: |
| python scripts/generate_degradation_gallery.py |
| """ |
|
|
| from __future__ import annotations |
|
|
| import sys |
| from pathlib import Path |
|
|
| sys.path.insert(0, str(Path(__file__).resolve().parents[1])) |
|
|
| from src.config.loader import load_config |
| from src.generator import GeneratorContext |
| from src.visualization.visualizer import save_visualization |
|
|
| SIGNATURE_EFFECTS = [ |
| "rotation", "stroke_dilation", |
| "ink_fading", "ink_bleed", "motion_blur", "gaussian_blur", |
| ] |
| DOCUMENT_EFFECTS = [ |
| "perspective", "contrast", "gamma", "coffee_stain", "stain", |
| "shadow", "crumple", "discoloration", "page_tilt", "dust", "scanner_streaks", |
| "scanner_noise", "salt_pepper", "jpeg_compression", |
| ] |
| ALL_EFFECTS = SIGNATURE_EFFECTS + DOCUMENT_EFFECTS |
|
|
| OUT_ROOT = Path(__file__).resolve().parents[1] / "degradation" |
| SAMPLES_PER_EFFECT = 5 |
| TEMPLATE_ID = "bank_doc_05" |
| FORCED_STRENGTH = 0.75 |
|
|
|
|
| def build_overrides(target_effect: str) -> list[str]: |
| overrides = [ |
| f"document.template={TEMPLATE_ID}", |
| "placement.rotation_deg.min=0", "placement.rotation_deg.max=0", |
| "placement.scale_factor.min=0.85", "placement.scale_factor.max=0.85", |
| "placement.x_offset=0", "placement.y_offset=0", |
| "placement.misplacement_probability=0.0", |
| "generation.run_validation=false", |
| ] |
| for name in ALL_EFFECTS: |
| is_target = name == target_effect |
| overrides.append(f"augmentation.{name}.enabled={'true' if is_target else 'false'}") |
| overrides.append(f"augmentation.{name}.probability={'1.0' if is_target else '0.0'}") |
| overrides.append(f"augmentation.{target_effect}.strength={FORCED_STRENGTH}") |
| return overrides |
|
|
|
|
| def main() -> None: |
| OUT_ROOT.mkdir(parents=True, exist_ok=True) |
| for effect in ALL_EFFECTS: |
| out_dir = OUT_ROOT / effect |
| out_dir.mkdir(parents=True, exist_ok=True) |
|
|
| config = load_config(overrides=build_overrides(effect)) |
| ctx = GeneratorContext(config) |
|
|
| for i in range(1, SAMPLES_PER_EFFECT + 1): |
| metadata, final_image, _ = ctx.run_sample(i) |
| save_visualization(out_dir / f"sample_{i}.png", final_image, metadata) |
|
|
| print(f"[{effect}] wrote {SAMPLES_PER_EFFECT} samples -> {out_dir}") |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
