Fix bug about split_model = False

gustavz / realtime_object_detection

Plug and Play Real-Time Object Detection App with Tensorflow and OpenCV

MIT License

281 stars 148 forks source link

Fix bug about split_model = False #9

Closed fenniewxy closed 6 years ago

fenniewxy commented 6 years ago

when split_model = False, there maybe an error like " Traceback (most recent call last): File "object_detection.py", line 314, in main() File "object_detection.py", line 310, in main detection(graph, category, score, expand) File "object_detection.py", line 297, in detection gpu_worker.stop() UnboundLocalError: local variable 'gpu_worker' referenced before assignment "

gustavz commented 6 years ago

thanks!