Introduce diffusion-inpaint restoration engine in HVideoTool: added support for a new restoration method that regenerates masked regions via an external SwarmUI server, requiring YOLO detections for mask creation. Updated configuration management to include diffusion parameters, enhanced the UI for engine selection, and improved documentation in README and CLAUDE.md to guide users on the new functionality.

This commit is contained in:
Leonid Pershin
2026-06-08 06:21:44 +03:00
parent 8a366ed43d
commit 15f89b395d
14 changed files with 903 additions and 70 deletions
+16
View File
@@ -75,3 +75,19 @@ class AppConfig:
# (instead of the original mosaic frames) for stronger temporal coherence. Slightly
# out-of-distribution for the net (trained on mosaic windows) — toggle in the dialog.
dm_feed_restored: bool = True
# --- diffusion-inpaint restoration ("diffusion" restorer) ---
# Regenerates the masked (detected) regions via an external diffusion server. Needs
# YOLO detections for the mask; the model runs out-of-process (no torch dep here).
# The backend is pluggable; only SwarmUI is wired so far.
diff_backend: str = "swarmui" # only "swarmui" implemented
diff_url: str = "http://localhost:7801" # SwarmUI server base URL
diff_model: str | None = None # checkpoint name as the server knows it
diff_prompt: str = ""
diff_negative: str = ""
diff_steps: int = 30
diff_cfg: float = 7.0
diff_denoise: float = 1.0 # 0..1, 1 = fully regenerate under the mask
diff_seed: int = -1 # -1 = random each call
diff_mask_dilate: int = 4 # px to grow the mask before inpaint
diff_mask_blur: int = 8 # px feather of the mask edge