PSPNet-tensorflow
PSPNet-tensorflow copied to clipboard
How to better do batch inference on multiple images
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 i'm also new to tf and face the same problem, have you solve it? could you show me how to do this?
Hi @Xziguoxu I do not have a solution yet
Hey @jin-zhe did you find a solution? Thanks.
Hi @Xziguoxu I do not have a solution yet
hi, everyone. did you find a solution now?