datitran / object_detector_app

Real-Time Object Recognition App with Tensorflow and OpenCV
https://medium.com/towards-data-science/building-a-real-time-object-recognition-app-with-tensorflow-and-opencv-b7a2b4ebdc32
MIT License
1.3k stars 745 forks source link

Performance issue in /object_detection/eval_util.py (by P3) #57

Open DLPerf opened 3 years ago

DLPerf commented 3 years ago

Hello! I've found a performance issue in /object_detection/eval_util.py: sess = tf.Session(master, graph=tf.get_default_graph())(here) is defined in the function run_checkpoint_once(here) which is repeatedly called in the loop while True(here).

tf.Session being defined repeatedly could lead to incremental overhead. If you define tf.Session out of the loop and pass tf.Session as a parameter to the loop, your program would be much more efficient. Here is the Stack Overflow post to support it.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

DLPerf commented 3 years ago

Hello, I'm looking forward to your reply~