VSGAN-tensorrt-docker
Using image super resolution models with vapoursynth and speeding them up with TensorRT. Using NVIDIA/Torch-TensorRT combined with rlaphoenix/VSGAN. This repo makes the usage of tiling and ESRGAN models very easy. Models can be found on the wiki page. Further model architectures are planned to be added later on.
Currently working:
- ESRGAN
- RealESRGAN (adjust model load manually in
inference.py, settings wont be adjusted automatically currently)
Usage:
# install docker, command for arch
yay -S docker nvidia-docker nvidia-container-toolkit
# Put the dockerfile in a directory and run that inside that directory
docker build -t vsgan_tensorrt:latest .
# run with a mounted folder
docker run --privileged --gpus all -it --rm -v /home/Desktop/tensorrt:/workspace/tensorrt vsgan_tensorrt:latest
# you can use it in various ways, ffmpeg example
vspipe --y4m inference.py - | ffmpeg -i pipe: example.mkv
If docker does not want to start, try this before you use docker:
# fixing docker errors
systemctl start docker
sudo chmod 666 /var/run/docker.sock
Windows is mostly similar, but the path needs to be changed slightly:
Example for C://path
docker run --privileged --gpus all -it --rm -v //c/path:/workspace/tensorrt vsgan_tensorrt:latest
If you don't want to use docker, vapoursynth install commands are here and a TensorRT example is here.
Set the input video path in inference.py and access videos with the mounted folder.
It is also possible to directly pipe the video into mpv, but you most likely wont be able to archive realtime speed. Change the mounted folder path to your own videofolder and use the mpv dockerfile instead. If you use a very efficient model, it may be possible on a very good GPU. Only tested in Manjaro.
yay -S pulseaudio
# i am not sure if it is needed, but go into pulseaudio settings and check "make pulseaudio network audio devices discoverable in the local network" and reboot
# start docker
docker run --rm -i -t \
--network host \
-e DISPLAY \
-v /home/Schreibtisch/test/:/home/mpv/media \
--ipc=host \
--privileged \
--gpus all \
-e PULSE_COOKIE=/run/pulse/cookie \
-v ~/.config/pulse/cookie:/run/pulse/cookie \
-e PULSE_SERVER=unix:${XDG_RUNTIME_DIR}/pulse/native \
-v ${XDG_RUNTIME_DIR}/pulse/native:${XDG_RUNTIME_DIR}/pulse/native \
vsgan_tensorrt:latest
# run mpv
vspipe --y4m inference.py - | mpv -
![[SUGGESTION] per-scene processing](https://avatars.githubusercontent.com/u/171307?v=4)
21 Nov 21, 2022
23 Dec 20, 2022
146 Nov 29, 2022
5 Apr 22, 2022
33 Jan 04, 2023
2 Aug 11, 2022
3 Aug 24, 2021
1.8k Dec 21, 2022
491 Jan 03, 2023
116 Jan 05, 2023
232 Dec 31, 2022
31 Dec 06, 2022
2 Dec 14, 2022
75 Dec 13, 2022
535 Nov 15, 2022
60 Dec 07, 2022
100 Nov 15, 2022
334 Dec 23, 2022
22 Dec 11, 2022
3 Mar 01, 2022