Implement DeepMosaics restoration engine in HVideoTool: updated configuration options, integrated into the UI, and enhanced documentation in README and CLAUDE.md. The restoration process now supports both inpainting and generative models, with necessary setup instructions included.
This commit is contained in:
@@ -35,14 +35,42 @@
|
||||
|
||||
Кнопка **«Расцензурить кадр»** восстанавливает найденные области на текущем кадре,
|
||||
**«Показать оригинал/результат»** переключает вид, **«Сохранить результат»** пишет
|
||||
`<имя>_restored.jpg` (в активную коллекцию или рядом с кадром).
|
||||
`<имя>_restored.jpg` (в активную коллекцию или рядом с кадром). Движок выбирается в
|
||||
меню **Файл → Движок восстановления…**.
|
||||
|
||||
> ⚠️ Сейчас движок восстановления — **инпейнт (cv2)**: он *заполняет* область по
|
||||
> окружению, но **не реконструирует** скрытые детали (замазывает, а не раскрывает).
|
||||
> Настоящее восстановление мозаики требует генеративной модели (DeepMosaics или
|
||||
> [LADA](https://github.com/ladaapp/lada) на BasicVSR++) и **GPU NVIDIA/CUDA**; на
|
||||
> аниме качество таких моделей ограничено. Движок подключается через интерфейс
|
||||
> `core/restore/base.Restorer` (`build_restorer`) — инпейнт сейчас стоит заглушкой.
|
||||
Два движка:
|
||||
|
||||
- **Инпейнт (cv2)** — по умолчанию, без модели и GPU. ⚠️ *Заполняет* область по
|
||||
окружению, но **не реконструирует** скрытые детали (замазывает, а не раскрывает).
|
||||
- **DeepMosaics** — реальное генеративное удаление мозаики. Требует **GPU NVIDIA/CUDA**
|
||||
(на CPU очень медленно) и отдельной установки. На аниме качество ограничено
|
||||
(модели обучены на реальном видео).
|
||||
|
||||
### Настройка DeepMosaics
|
||||
|
||||
```powershell
|
||||
git clone https://github.com/HypoX64/DeepMosaics
|
||||
# установите зависимости DeepMosaics (см. его README; нужен torch с CUDA)
|
||||
# скачайте веса (clean_youknow_resnet_9blocks.pth + mosaic_position.pth) в pretrained_models/mosaic
|
||||
```
|
||||
|
||||
Затем в приложении: **Файл → Движок восстановления… → DeepMosaics**, укажите папку
|
||||
DeepMosaics (с `deepmosaic.py`), файл весов и GPU id (`-1` = CPU). Приложение вызывает
|
||||
DeepMosaics на текущем кадре и показывает результат.
|
||||
|
||||
> **Важно:** для покадрового режима берите **картиночную** модель
|
||||
> `clean_youknow_resnet_9blocks.pth`. Видеомодель `clean_youknow_video.pth` (BVDNet)
|
||||
> покадрово **не работает** — ей нужен соседний кадр. Если кадр без мозаики, движок
|
||||
> вернёт его без изменений.
|
||||
>
|
||||
> DeepMosaics 2021 года рассчитан на старые версии (`torch 1.7`, `numpy 1.19`).
|
||||
> Проверено: на современных `torch 2.x`/`numpy 2.x` картиночная модель запускается
|
||||
> (CPU ~7 с/кадр), но если столкнётесь с несовместимостью — заведите для DeepMosaics
|
||||
> отдельное окружение по его `requirements.txt` и укажите его `python.exe` в диалоге.
|
||||
|
||||
> Движок подключается через интерфейс `core/restore/base.Restorer` (`build_restorer`).
|
||||
> [LADA](https://github.com/ladaapp/lada) (BasicVSR++, лучшее качество на реальном
|
||||
> видео, но видеомодель) пока не подключён.
|
||||
|
||||
## Что НЕ делает (осознанно вне области задачи)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user