add set resolution

Author: tobybreckon

contour_edges.py

@@ -38,6 +38,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(
     'video_file',
     metavar='video_file',
@@ -107,6 +113,16 @@ def nothing(x):
     sobel_size = 3  # greater than 7 seems to crash
     cv2.createTrackbar("sobel size", window_name2, sobel_size, 7, nothing)
 
+    # 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])
+
+    print("INFO: input resolution : (",
+          int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), "x",
+          int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)), ")")
+
     while (keep_processing):
 
         # if video file successfully open then read frame from video