Данный репозиторий можно использовать для увеличения качества видео и его разрешения с помощью нейронных сетей от Nvidia.
Выполнено на основе: MAXINE-VFX-SDK
Код в репозитории работает с помощью команд интерфейса CLI.
Доступны следующие преобразования видео:
- Upscale - увеличивает разрешение входного видео. Во время работы применяются по умолчанию также алгоритмы:
- Artifact Reduction - эффект исключения искажений;
- Super Resolution - эффект улучшения деталей и четкости;
- Denoising - эффект удаления шума.
Каждый эффект можно также применять отдельно ко всему видео. Это позволяет выстраивать цепочки последовательных преобразований в желаемом порядке.
Чтобы применить каждый эффект в желаемом порядке за одно преобразование (в потоке) необходимо декодировать видео, выполнить необходимые преобразования каждого кадра, а затем энкодером преобразовывать их в выходное видео.
Одновременно Artifact Reduction и Super Resolution доступны по умолчанию только с Upscale.
Библиотека работает на компьютерах с видеокартами, имеющими архитектуру Turing или Ampere от NVIDIA.
Поддерживается 64-разрядная версия Windows 10 или более поздняя.
Графический драйвер NVIDIA для Windows должен быть 465.89 или более поздней версии.
- Установить SDK:
- Установить Python >= 3.8.
- В командной строке Windows:
- Скачать текущий репозиторий: git clone https://github.com/uralskayamariya/MAXINE-VFX-SDK_video_stream.git
- Перейти в папку текущего репозитория: cd MAXINE-VFX-SDK_video_stream
- Запустить скрипт run.py: python run.py
- Ввести данные, которые запрашивает программа:
- Выбрать алгоритм:
- 1-Upscale (включает Artifact Reduction, Super Resolution) (Допускается масштаб от размера входного изображения по вертикали 1.3333x, 1.5x, 2x, 3x или 4x.)
- 2-Denoising
- 3-Artifact Reduction
- 4-Super Resolution (Допускается масштаб от размера входного изображения по вертикали 1.3333x, 1.5x, 2x, 3x или 4x.)
- Указать путь ко входному видео. Если положить видео в папку input в текущем каталоге, можно указать только его имя, например: input_video.mp4
- Указать путь к выходному видео. Если хотите сохранить видео в папку output в текущем каталоге, можно указать только его имя, например: output_video.mp4. Папка output должна быть создана заранее.
- Выбрать алгоритм:
- NVIDIA Video Effects SDK Programming Guide
- NVIDIA Video Effects SDK System Guide
- NvCVImage API Guide
PDF версии доступны по следующим ссылкам: