Skip to content

Commit

Permalink
bug fixed (#196)
Browse files Browse the repository at this point in the history
Co-authored-by: Negar <negar.heidari@eng.au.dk>
Co-authored-by: ad-daniel <44834743+ad-daniel@users.noreply.github.com>
  • Loading branch information
3 people authored Jan 14, 2022
1 parent 5cb4c3c commit 49f7cd9
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ def draw_preds(frame, preds: Dict):
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--onnx", help="Use ONNX", default=False, action="store_true")
parser.add_argument("--device", help="Device to use (cpu, cuda)", type=str, default="cuda")
parser.add_argument("--device", help="Device to use (cpu, cuda)", type=str, default="cpu")
parser.add_argument("--accelerate", help="Enables acceleration flags (e.g., stride)", default=False,
action="store_true")
parser.add_argument('--video', default=0,
Expand Down Expand Up @@ -224,5 +224,4 @@ def draw_preds(frame, preds: Dict):
break

print("Average inference fps: ", avg_fps)
image_provider.release()
cv2.destroyAllWindows()

0 comments on commit 49f7cd9

Please sign in to comment.