SwinIR
SwinIR copied to clipboard
JingyunLiang
The noise removal command eats up my entire RAM and then gets killed
I have 64GB RAM and 30GB swap but when I run the following command:
python main_test_swinir.py --task colour_dn --noise 15 --model_path model_zoo/swinir/005_colorDN_DFWB_s128w8_SwinIR-M_noise15.pth --folder_gt testsets/own
on a few pictures with 12megapixel first my ram is filled up, then the swap and then the program is killed by the system. I haven't tested what happens with smaller images, but I can't imagine it necessarily takes that much RAM.
add to the startup command: --tile 400
it will turn out like this: python main_test_swinir.py --task colour_dn --noise 15 --model_path model_zoo/swinir/005_colorDN_DFWB_s128w8_SwinIR-M_noise15.pth --folder_gt testsets/own --tile 400
tell me, did you manage to solve the problem later?
this option processes the image in chunks, small tiles and not all at once, it saves memory and there is no message about low RAM
I do not know exactly what this parameter does inside the program, I know that this parameter helps to process large images without reporting a lack of memory. you need to ask the developer in more detail, but he does not seem to get in touch and is silent.
