Parse command line args

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():