forked from goktug97/TargetPersonTracker
-
Notifications
You must be signed in to change notification settings - Fork 0
/
example.py
30 lines (25 loc) · 896 Bytes
/
example.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import tracker
import cv2
import utils
import numpy as np
class Example(tracker.Tracker):
"""Example usage of Tracker class."""
def __init__(self, args):
"""Initiliaze tracker and image source."""
tracker.Tracker.__init__(self, args)
# This is the default
self._cap = cv2.VideoCapture(self.args.input)
# Camera Settings
self._cap.set(cv2.CAP_PROP_FRAME_WIDTH, self.args.camera_width)
self._cap.set(cv2.CAP_PROP_FRAME_HEIGHT, self.args.camera_height)
if (self.args.camera_fps):
self._cap.set(cv2.CAP_PROP_FPS, self.args.camera_fps)
def get_frame(self):
"""Overload get_frame function."""
ret, self.frame = self._cap.read()
return ret, self.frame
if __name__ == '__main__':
from options import args
tracker = Example(args)
tracker.initiliaze_target()
tracker.run()