Inference is causing Out of Memory error (even for V100)

[wind-surfer]

Hello,
I tried running 'main_test_fbcnn_color.py' on a real JPEG image using one 16 GB V100 but the code threw ´Out of Memory´ error. Any idea on how to use this code with large images, say 12 MPix or more?

[jiaxi-jiang]

Hi, thanks for your interest!

You can try to either run on the CPU or split the large image into smaller ones, e.g. based on the code here:https://github.com/cszn/KAIR/blob/master/utils/utils_model.py#L172

[wind-surfer]

Hello,
Thank you for your prompt reply. Do you think splitting->inference->merging will produce any artifacts at the edges of the image tile?

[jiaxi-jiang]

Hi, I don't think so. I once tested large images using this method and did not see produced artifacts at the edges.

[wind-surfer]

Sure. Thank you for your prompt response.