Invertible-Image-Rescaling icon indicating copy to clipboard operation
Invertible-Image-Rescaling copied to clipboard

why is so slow? in one V100 GPU, 10 second per frame?

Open xiaoxiongli opened this issue 3 years ago • 2 comments

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?

xiaoxiongli avatar Apr 13 '21 08:04 xiaoxiongli

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?

pkuxmq avatar Apr 13 '21 09:04 pkuxmq

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.

xiaoxiongli avatar May 28 '21 07:05 xiaoxiongli