23 lines
929 B
Python
23 lines
929 B
Python
"""Restorer factory: build a Restorer by name.
|
|
|
|
Currently only the cv2 inpaint baseline is wired. Generative engines
|
|
(DeepMosaics / LADA BasicVSR++) are placeholders — they need model weights and a
|
|
CUDA GPU, and raise a clear, actionable error until integrated. See README.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from .base import Restorer
|
|
from .inpaint import InpaintRestorer
|
|
|
|
|
|
def build_restorer(name: str = "inpaint", model_path: str | None = None) -> Restorer:
|
|
if name == "inpaint":
|
|
return InpaintRestorer()
|
|
if name in ("deepmosaics", "lada"):
|
|
raise ValueError(
|
|
"Генеративное восстановление пока не подключено.\n"
|
|
"Нужна модель (DeepMosaics / LADA) и GPU (CUDA). См. README → Восстановление."
|
|
)
|
|
raise ValueError(f"Неизвестный режим восстановления: {name!r}")
|