search

pkuxmq / Invertible-Image-Rescaling

[ECCV 2020, IJCV 2022] Invertible Image Rescaling
Apache License 2.0
631 stars 86 forks source link

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

Open xiaoxiongli opened 3 years ago

xiaoxiongli commented 3 years ago

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?

pkuxmq commented 3 years ago

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?

xiaoxiongli commented 3 years ago

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.