GPU memory while increasing image resolution

[primecai]

Hello,

I am attempting to run the model on higher resolution(1024x768), but I got CUDA out of memory error on a 12gb GPU. If I downscale the images to half the size(512x384), it can run perfectly while taking at most 80% of the GPU memory. I thought most of the memory is because of the stored keypoints and descriptors so I decreased the batch size from 8 to 1(while decreasing to 2 should be enough), but it still cannot fit in 12gb memory. I'm wondering what exactly is taking so much memory while increasing the image resolution?

Many thanks.

[RaresAmbrus]

Hi, that's strange, I'm not sure what's causing that high memory usage, especially with batch size 1. Have you had any luck figuring this out?