From 69caa21c01ef474a6df8d7d333b05ea84b6adb3e Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 5 Jul 2024 21:42:43 +0200 Subject: [PATCH] Parse command line args --- ocv_corner_tracker.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/ocv_corner_tracker.py b/ocv_corner_tracker.py index f96e7ed..8cd8fb1 100644 --- a/ocv_corner_tracker.py +++ b/ocv_corner_tracker.py @@ -1,6 +1,7 @@ import cv2 import numpy as np import imutils as im +import argparse from util import image_crop, to_rect, bbox_add_position, bbox_center, bbox_round @@ -307,10 +308,14 @@ class CornerTracker: if __name__ == '__main__': + # Parse command line args + parser = argparse.ArgumentParser(description="Gearbox Tracker") + parser.add_argument("filename") + args = parser.parse_args() + video = cv2.VideoCapture(args.filename) + + # Let's go colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 0, 255), (0, 255, 255), (255, 255, 255)] -# video = cv2.VideoCapture('./data/spindle_multi_black/%04d.png') -# video = cv2.VideoCapture('./data/spindle_jet_12/0013/results_20240621_165000/gear_4/Distance_33.7/video/Distance_33.7_5fps.mp4') - video = cv2.VideoCapture('./data/spindle_jet_12/0020/Distance_70/video/Distance_70_5fps.mp4') tracker_count = 0 tracker_list = [] if video.isOpened():