Python cv2.TrackerKCF_create() Examples
The following are 10
code examples of cv2.TrackerKCF_create().
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example.
You may also want to check out all available functions/classes of the module
cv2
, or try the search function
.
.
Example #1
| Source File: test_blob.py From ivy with MIT License | 7 votes |
|
def test_blob_update():
_bounding_box = [1, 1, 4, 4]
_type = 'car'
_confidence = 0.99
_tracker = cv2.TrackerKCF_create()
blob = Blob(_bounding_box, _type, _confidence, _tracker)
_new_bounding_box = [2, 2, 5, 5]
_new_type = 'bus'
_new_confidence = 0.35
_new_tracker = cv2.TrackerCSRT_create()
blob.update(_new_bounding_box, _new_type, _new_confidence, _new_tracker)
assert blob.bounding_box == _new_bounding_box
assert blob.type == _new_type
assert blob.type_confidence == _new_confidence
assert blob.tracker == _new_tracker
Example #2
| Source File: ground-control.py From aws-builders-fair-projects with Apache License 2.0 | 6 votes |
|
def predictionCallback(self, client, userdata, message):
print('<< predictionCallback() >>')
data = json.loads(message.payload.decode())
if len(data['prediction']) > 0 and data['prediction'][0][0] > -1:
self.last_confidence = round(data['prediction'][0][2]*100,2)
self.last_detected_class = int(data['prediction'][0][0])
x1 = data['prediction'][0][2]*self.IMAGE_WIDTH
y1 = data['prediction'][0][3]*self.IMAGE_HEIGHT
x2 = data['prediction'][0][4]*self.IMAGE_WIDTH
y2 = data['prediction'][0][5]*self.IMAGE_HEIGHT
w = x2-x1
h = y2-y1
print("({},{},{},{})".format(x1,x2,y1,y2))
print("({},{})".format(w,h))
self.initBB = (int(x1), int(y1), int(w), int(h))
# self.tracker = cv2.TrackerMOSSE_create()
# self.tracker = cv2.TrackerKCF_create()
self.tracker = cv2.TrackerCSRT_create()
self.tracker.init(self.inferenceFrame, self.initBB)
self.trackerInitialized = True
Example #3
| Source File: utils.py From face_landmark_dnn with MIT License | 5 votes |
|
def __init__(self, args):
self.args = args
# self.tracker = cv2.TrackerMedianFlow_create()
self.tracker = cv2.TrackerKCF_create()
Example #4
| Source File: utils.py From face_landmark_dnn with MIT License | 5 votes |
|
def initTracker(self, img, bbox):
bbox_xywh = self.xyxy2xywh(bbox)
self.tracker = cv2.TrackerKCF_create()
# self.tracker = cv2.TrackerMedianFlow_create()
ok = self.tracker.init(img, bbox_xywh)
return ok
Example #5
| Source File: main_auto.py From OpenLabeling with Apache License 2.0 | 5 votes |
|
def call_tracker_constructor(self, tracker_type):
# -- TODO: remove this if I assume OpenCV version > 3.4.0
if int(self.major_ver == 3) and int(self.minor_ver) < 3:
tracker = cv2.Tracker_create(tracker_type)
# --
else:
if tracker_type == 'CSRT':
tracker = cv2.TrackerCSRT_create()
elif tracker_type == 'KCF':
tracker = cv2.TrackerKCF_create()
elif tracker_type == 'MOSSE':
tracker = cv2.TrackerMOSSE_create()
elif tracker_type == 'MIL':
tracker = cv2.TrackerMIL_create()
elif tracker_type == 'BOOSTING':
tracker = cv2.TrackerBoosting_create()
elif tracker_type == 'MEDIANFLOW':
tracker = cv2.TrackerMedianFlow_create()
elif tracker_type == 'TLD':
tracker = cv2.TrackerTLD_create()
elif tracker_type == 'GOTURN':
tracker = cv2.TrackerGOTURN_create()
return tracker
Example #6
| Source File: opencv_cftracker.py From pyCFTrackers with MIT License | 5 votes |
|
def init(self, first_frame, bbox):
if self.name == 'KCF':
self.tracker = cv2.TrackerKCF_create()
elif self.name == 'MOSSE':
self.tracker = cv2.TrackerMOSSE_create()
elif self.name == 'CSRDCF':
self.tracker = cv2.TrackerCSRT_create()
else:
raise NotImplementedError
self.tracker.init(first_frame, bbox)
Example #7
| Source File: test_blob.py From ivy with MIT License | 5 votes |
|
def test_blob_creation():
_bounding_box = [1, 1, 4, 4]
_type = 'car'
_confidence = 0.99
_tracker = cv2.TrackerKCF_create()
blob = Blob(_bounding_box, _type, _confidence, _tracker)
assert isinstance(blob, Blob), 'blob is an instance of class Blob'
assert blob.bounding_box == _bounding_box
assert blob.type == _type
assert blob.type_confidence == _confidence
assert isinstance(blob.tracker, cv2.Tracker), 'blob tracker is an instance of OpenCV Tracker class'
Example #8
| Source File: tracker.py From ivy with MIT License | 5 votes |
|
def _kcf_create(bounding_box, frame):
'''
Create an OpenCV KCF Tracker object.
'''
tracker = cv2.TrackerKCF_create()
tracker.init(frame, tuple(bounding_box))
return tracker
Example #9
| Source File: main.py From OpenLabeling with Apache License 2.0 | 4 votes |
|
def call_tracker_constructor(self, tracker_type):
if tracker_type == 'DASIAMRPN':
tracker = dasiamrpn()
else:
# -- TODO: remove this if I assume OpenCV version > 3.4.0
if int(self.major_ver == 3) and int(self.minor_ver) < 3:
#tracker = cv2.Tracker_create(tracker_type)
pass
# --
else:
try:
tracker = cv2.TrackerKCF_create()
except AttributeError as error:
print(error)
print('\nMake sure that OpenCV contribute is installed: opencv-contrib-python\n')
if tracker_type == 'CSRT':
tracker = cv2.TrackerCSRT_create()
elif tracker_type == 'KCF':
tracker = cv2.TrackerKCF_create()
elif tracker_type == 'MOSSE':
tracker = cv2.TrackerMOSSE_create()
elif tracker_type == 'MIL':
tracker = cv2.TrackerMIL_create()
elif tracker_type == 'BOOSTING':
tracker = cv2.TrackerBoosting_create()
elif tracker_type == 'MEDIANFLOW':
tracker = cv2.TrackerMedianFlow_create()
elif tracker_type == 'TLD':
tracker = cv2.TrackerTLD_create()
elif tracker_type == 'GOTURN':
tracker = cv2.TrackerGOTURN_create()
return tracker
Example #10
| Source File: vis_tracker.py From iou-tracker with MIT License | 4 votes |
|
def __init__(self, tracker_type, bbox, img, keep_height_ratio=1.):
""" Wrapper class for various visual trackers."
Args:
tracker_type (str): name of the tracker. either the ones provided by opencv-contrib or KCF2 for a different
implementation for KCF (requires https://github.com/uoip/KCFcpp-py-wrapper)
bbox (tuple): box to initialize the tracker (x1, y1, x2, y2)
img (numpy.ndarray): image to intialize the tracker
keep_height_ratio (float, optional): float between 0.0 and 1.0 that determines the ratio of height of the
object to track to the total height of the object for visual tracking.
"""
if tracker_type == 'KCF2' and not KCF:
tracker_type = 'KCF'
if not VisTracker.kcf2_warning_printed:
print("[warning] KCF2 not available, falling back to KCF. please see README.md for further details")
VisTracker.kcf2_warning_printed = True
self.tracker_type = tracker_type
self.keep_height_ratio = keep_height_ratio
if tracker_type == 'BOOSTING':
self.vis_tracker = cv2.TrackerBoosting_create()
elif tracker_type == 'MIL':
self.vis_tracker = cv2.TrackerMIL_create()
elif tracker_type == 'KCF':
self.vis_tracker = cv2.TrackerKCF_create()
elif tracker_type == 'KCF2':
self.vis_tracker = KCF.kcftracker(False, True, False, False) # hog, fixed_window, multiscale, lab
elif tracker_type == 'TLD':
self.vis_tracker = cv2.TrackerTLD_create()
elif tracker_type == 'MEDIANFLOW':
self.vis_tracker = cv2.TrackerMedianFlow_create()
elif tracker_type == 'GOTURN':
self.vis_tracker = cv2.TrackerGOTURN_create()
elif tracker_type == 'NONE': # dummy tracker that does nothing but fail
self.vis_tracker = None
self.ok = False
return
else:
raise ValueError("Unknown tracker type '{}".format(tracker_type))
y_max = img.shape[0] - 1
x_max = img.shape[1] - 1
#
bbox = list(bbox)
bbox[0] = max(0, min(bbox[0], x_max))
bbox[2] = max(0, min(bbox[2], x_max))
bbox[1] = max(0, min(bbox[1], y_max))
bbox[3] = max(0, min(bbox[3], y_max))
bbox = [bbox[0], bbox[1], bbox[2] - bbox[0], bbox[3] - bbox[1]] # x1, y1, x2, y2 -> x1, y1, w, h
bbox[3] *= self.keep_height_ratio
if self.tracker_type == 'KCF2':
self.vis_tracker.init(bbox, img)
self.ok = True
else:
self.ok = self.vis_tracker.init(img, tuple(bbox))
pass
