search

bloc97 / Anime4K

A High-Quality Real Time Upscaler for Anime Video
https://bloc97.github.io/Anime4K/
MIT License
17.98k stars 1.34k forks source link

Add Anime4K-Pytorch #233

Closed Fannovel16 closed 4 months ago

Fannovel16 commented 4 months ago

https://colab.research.google.com/drive/11xAn4fyAUJPZOjrxwnL2ipl_1DGGegkB Based on @kato-megumi convert script: https://gist.github.com/kato-megumi/44e52b4cc0e082e94d452a7df04243e0 Support models: ['Upscale_S', 'Upscale_M', 'Upscale_L', 'Upscale_VL', 'Upscale_UL', 'Upscale_Denoise_S', 'Upscale_Denoise_M', 'Upscale_Denoise_L', 'Upscale_Denoise_VL', 'Upscale_Denoise_UL', 'Restore_S', 'Restore_L', 'Restore_VL', 'Restore_UL', 'Restore_Soft_S', 'Restore_Soft_M', 'Restore_Soft_L', 'Restore_Soft_VL', 'Restore_Soft_UL'] This notebook also supports fp16 and TensorRT

Fannovel16 commented 4 months ago

Archived 16FPS on Colab T4 on Preset A (HQ) btw

bloc97 commented 4 months ago

LGTM, thanks!

bloc97 commented 4 months ago

Archived 16FPS on Colab T4 on Preset A (HQ) btw

Interesting that it would be so low on a Tesla T4 (which is basically a 3060 with more VRAM), I would have expected at least 80FPS+...

Fannovel16 commented 4 months ago

Interesting that it would be so low on a Tesla T4 (which is basically a 3060 with more VRAM), I would have expected at least 80FPS+...

That is only 720p with batch size 1. Idk if MPV/libplacebo processes frame-by-frame or does batch processing.