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)
32 Dec 20, 2022
333 Dec 29, 2022
1 Dec 10, 2021
3 Aug 24, 2022
897 Jan 05, 2023
5 Jul 31, 2022
2 Oct 31, 2022
17 Nov 22, 2022
43 Dec 05, 2022
6 Jan 14, 2022
45 Dec 19, 2022
842 Jan 04, 2023
53 Dec 28, 2022
12 Jul 08, 2022
12 Jul 22, 2022
69 Dec 31, 2022
601 Nov 23, 2022
1 Jun 29, 2022
37 Oct 09, 2022
75 Dec 16, 2022