MIRNet-Keras icon indicating copy to clipboard operation
MIRNet-Keras copied to clipboard

SR won't work on real world images

Open ofirkris opened this issue 5 years ago • 5 comments

Hi, thanks for sharing this implementation. I'm researching SR for real world cases - and get out of memory issues with an empty 16GB P100 GPU

2020-11-26 09:03:19.928246: W tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 2488320000 exceeds 10% of free system memory.

I've tested SR with 1200px and this crashes each time

ofirkris avatar Nov 26 '20 09:11 ofirkris

hi, how much memory do you have in your GPU? The number of this network is quite huge (34M) so that kind of big image will cause the OOM like that. If you have any idea on reducing the number or params, please feel free to make a pull request!

tuvovan avatar Nov 26 '20 09:11 tuvovan

I have 16GB of GPU (P100) The implementation here - https://github.com/swz30/MIRNet does work on such images - but the Super-Resolution output effect is not good compared to the input image.

Is Mirnet currently still the SoTa compared to "Deep Plug-and-Play Image Restoration" https://github.com/cszn/DPIR

P.S - the link to Google Drive - for weights of the lighting model isn't correct.

ofirkris avatar Nov 26 '20 09:11 ofirkris

let share to me the link to the image that you want to test. I will try with my machine.

And thank you for your contributions! :D

tuvovan avatar Nov 26 '20 09:11 tuvovan

And about SOTA or not: I would say that depends on the number provided by the original paper. I was not able to reach the number stated in the paper anyway.

tuvovan avatar Nov 26 '20 09:11 tuvovan

Thanks! I tried with this image https://i.ytimg.com/vi/agLZhd98Sus/maxresdefault.jpg

ofirkris avatar Nov 26 '20 15:11 ofirkris