jens/ocv_template_matching
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Parse command line args

Browse Source
This commit is contained in:
jens
2024-07-05 21:42:43 +02:00
parent a0002d91c0
commit 69caa21c01
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ocv_corner_tracker.py
+8 -3
View File
@@ -1,6 +1,7 @@
import cv2 import cv2
import numpy as np import numpy as np
import imutils as im import imutils as im
import argparse
from util import image_crop, to_rect, bbox_add_position, bbox_center, bbox_round from util import image_crop, to_rect, bbox_add_position, bbox_center, bbox_round
@@ -307,10 +308,14 @@ class CornerTracker:
if __name__ == '__main__': 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)] 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_count = 0
tracker_list = [] tracker_list = []
if video.isOpened(): if video.isOpened():