Invertible-Image-Rescaling
Invertible-Image-Rescaling copied to clipboard
why is so slow? in one V100 GPU, 10 second per frame?
Hi, @all, this is really good work, and I test it on DIV2k image, and 4 times SR from about resolution 400x500 --> 1600x2000, but is seems really slow: in one V100 GPU, it cost about 10 second per frame.
I wonder to know why is so slow, which part of the technology is slow? INN?
Hi, we have tested the models on one Tesla P100 GPU, when the HR size is 1920 x 1080, the downscaling and upscaling times for 2 times scale IRN_x2 are both around 345ms, and for 4 times scale IRN_x4 are both around 520ms. It should not be much longer for 1600 x 2000 images. I wonder how you calculate the time, e.g. if you take the calculation of PSNR and SSIM on cpu into consideration?
sorry, I just see your print log. and I write some code to test the infercen time, it is about 1s in a 2K image, I use IRN_x4 and P100.