The infeence time

[fasogbon]

trafficstars

Thank you for this great work! Is it normal that the inference time takes very long time for an image? It takes more than 7minutes of even much more.

And if I am right, the inference step don't use GPU, right?

[riaverma96]

Using GPU for inference sped up my inference time considerably.

[MuhammadAsadJaved]

Thank you for this great work! Is it normal that the inference time takes very long time for an image? It takes more than 7minutes of even much more.

And if I am right, the inference step don't use GPU, right?

Hi, if you sucessfuly run this project can you tell me about environment settings? i.e TensorFlow and other libraries versions?

[neonb88]

Using GPU for inference sped up my inference time considerably.

Is this a simple flag we can feed in (like --gpu yes), or do I have to go into the code, read & edit it?

[MaxMayya]

@riaverma96 this is interesting! can you share more details?

[mhudnell]

Using a NVIDIA Quadro P5000 gpu, my system takes ~4 sec per 405x720 px image

Is this a simple flag we can feed in (like --gpu yes), or do I have to go into the code, read & edit it?

I did not have to edit the code at all, just pip install tensorflow-gpu==1.15.0, and make sure you follow instructions on setting up tensorflow gpu support.

[MathamPollard]

but how do you get the gpu pre-training checkpoints? Dosen't the author only provide cpu pre-training checkpoints?