the speed of the demo

[yizi22]

hello, your project works really very well, but it takes 1 second to predict one image on GPU of 12G memory, I want to ask is it normal? thank you.

[yuanyuanli85]

For computation, it is normal. The computation comes from several places.

• The pre-trained model uses resnet101 as backbone
• The its input is 512x512.
• For testing, it uses merged result from rotate/flip/original image.

For memory consumption, don't need 12G that much. It is tensorflow's behavior to occupy all the memory.

[yizi22]

@yuanyuanli85 thank you for your reply. I set the predict batch_size to be 32 and it works much faster.