bottom-up-attention Required gpu memory

Hello I am trying to use the pretrained model to extract image features, I have GTX 1070(8 gb) and I get out of memory error when I use the network on one image, I suspect this is a caffe issue regarding memory mangament, what do you suggest to solve this issue without decreasing performance?

Mar 23 '18 07:03 ObadaAljabasini

@ObadaJabassini Have you solved this issue? I am facing the same problem.

Apr 18 '18 10:04 ZhuFengdaaa

There are some comments about how to get around the memory requirements in issue #3

Apr 18 '18 13:04 peteanderson80

@ObadaJabassini Have you solved this issue? I am facing the same problem.

hi, have you solved problem?

Jun 19 '21 08:06 maoyj2333

I use this code to extract feature from VG datasets, and I found it was caused by some images with big difference in aspect ratio, for example 281 * 500, faster rcnn will scale based on the shorter one, so making the larger one to large. this code may work on 11G gpu

Jun 21 '21 00:06 maoyj2333