add option to force resolution from camera

Author: tobybreckon

yolo.py

@@ -55,6 +55,12 @@
     type=float,
     help="rescale image by this factor",
     default=1.0)
+parser.add_argument(
+    "-s",
+    "--set_resolution",
+    type=int,
+    nargs=2,
+    help='override default camera resolution as H W')
 parser.add_argument(
     "-fs",
     "--fullscreen",
@@ -281,6 +287,12 @@ def getOutputsNames(net):
     # create window by name (as resizable)
     cv2.namedWindow(window_name, cv2.WINDOW_NORMAL)
 
+    # override default camera resolution
+
+    if (args.set_resolution is not None):
+        cap.set(cv2.CAP_PROP_FRAME_HEIGHT, args.set_resolution[1])
+        cap.set(cv2.CAP_PROP_FRAME_WIDTH, args.set_resolution[0])
+
     while (keep_processing):
 
         # start a timer (to see how long processing and display takes)