revamp dockerfile

This commit is contained in:
Blake Blackshear
2020-01-05 17:43:14 -06:00
parent cc2abe93a6
commit 10dc56f6ea
4 changed files with 39 additions and 101 deletions

View File

@@ -32,7 +32,7 @@ class PreppedQueueProcessor(threading.Thread):
frame = self.prepped_frame_queue.get()
# Actual detection.
frame['detected_objects'] = self.engine.DetectWithInputTensor(frame['frame'], threshold=0.2, top_k=5)
frame['detected_objects'] = self.engine.detect_with_input_tensor(frame['frame'], threshold=0.2, top_k=5)
self.fps.update()
self.avg_inference_speed = (self.avg_inference_speed*9 + self.engine.get_inference_time())/10