PSPNet-tensorflow icon indicating copy to clipboard operation
PSPNet-tensorflow copied to clipboard

How to better do batch inference on multiple images

Open jin-zhe opened this issue 7 years ago • 4 comments

Hi there! My apologies if this is a stupid question as I am still very new to TF. I am trying to use PSP50 (pre-trained on ADE20K) for inference on a list of images. Currently I am doing this via modifying inference.py by injecting tf.reset_default_graph() at the top of the inference code block which is iterated for every image. This feels very inefficient as it is recreating the net and loading the checkpoint for every image. Is there a more optimal way to go about doing this?

jin-zhe avatar Apr 10 '18 06:04 jin-zhe

@jin-zhe i'm also new to tf and face the same problem, have you solve it? could you show me how to do this?

Xziguoxu avatar May 28 '18 03:05 Xziguoxu

Hi @Xziguoxu I do not have a solution yet

jin-zhe avatar May 28 '18 09:05 jin-zhe

Hey @jin-zhe did you find a solution? Thanks.

jackdh avatar Nov 22 '18 10:11 jackdh

Hi @Xziguoxu I do not have a solution yet

hi, everyone. did you find a solution now?

Mickeyyyang avatar Mar 29 '20 06:03 Mickeyyyang