abdelrahman-gaber / tf2-object-detection-api-tutorial

Tensorflow 2 Object Detection API Tutorial. This tutorial will take you from installation, to running pre-trained detection model, and training your model with a custom dataset, then exporting it for inference.
MIT License
114 stars 52 forks source link

A way to find Inference speed #5

Open ALTAIR-VEGA opened 3 years ago

ALTAIR-VEGA commented 3 years ago

Is there a way to find the Inference speed of my .pb file ? with a modification to the files

abdelrahman-gaber commented 3 years ago

It is already done starting from this line: https://github.com/abdelrahman-gaber/tf2-object-detection-api-tutorial/blob/master/detect_objects.py#L22 finally we show FPS but first we get the inference time as show in previous link.