Merge pull request #16 from burningion/main

Make camera a parameter
NVIDIA-AI-IOT · Feb 9, 2024 · 9ae3d83 · 9ae3d83
2 parents 3d36c50 + 8c53152
commit 9ae3d83
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/tree_demo/tree_demo.py b/examples/tree_demo/tree_demo.py
@@ -38,9 +38,10 @@
     parser.add_argument("--image_quality", type=int, default=50)
     parser.add_argument("--port", type=int, default=7860)
     parser.add_argument("--host", type=str, default="0.0.0.0")
+    parser.add_argument("--camera", type=int, default=0)
     args = parser.parse_args()
 
-    CAMERA_DEVICE = 0
+    CAMERA_DEVICE = args.camera
     IMAGE_QUALITY = args.image_quality
 
     predictor = TreePredictor(