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

@@ -14,7 +14,7 @@ flattened_frame = np.expand_dims(frame, axis=0).flatten()
detection_times = []
for x in range(0, 1000):
objects = engine.DetectWithInputTensor(flattened_frame, threshold=0.1, top_k=3)
objects = engine.detect_with_input_tensor(flattened_frame, threshold=0.1, top_k=3)
detection_times.append(engine.get_inference_time())
print("Average inference time: " + str(statistics.mean(detection_times)))