def createTrackerByName(trackerType):
    if trackerType == tracker_types[0]:
        tracker = cv2.TrackerBoosting_create()
    elif trackerType == tracker_types[1]:
        tracker = cv2.TrackerMIL_create()
    elif trackerType == tracker_types[2]:
        tracker = cv2.TrackerKCF_create()
    elif trackerType == tracker_types[3]:
        tracker = cv2.TrackerTLD_create()
    elif trackerType == tracker_types[4]:
        tracker = cv2.TrackerMEDIANFLOW_create()
    elif trackerType == tracker_types[5]:
        tracker = cv2.TrackerMOSSE_create()
    elif trackerType == tracker_types[6]:
        tracker = cv2.TrackerCSRT_create()
    else:
        tracker = None
        print('Nome incorreto')
        print('Os rastreadores disponiveis são:')
        for t in tracker_types:
            print(t)

    return tracker
Ejemplo n.º 2
0
tracker_types = ['BOOSTING', 'MIL', 'KCF', 'TLD', 'MEDIANFLOW']
tracker_type = tracker_types[2]

if  int(minor_ver)>=3 and int(major_ver)>2:
    if tracker_type=='BOOSTING':
        tracker=cv.TrackerBoosting_create()

    if tracker_type=='MIL':
        tracker=cv.TrackerMIL_create()
    if tracker_type=='KCF':
        tracker=cv.TrackerKCF_create()
    if tracker_type=='TLD':
        tracker=cv.TrackerTLD_create()
    if tracker_type=='MEDIANFLOW':
        tracker=cv.TrackerMEDIANFLOW_create()

video= cv.VideoCapture(0) #'D:/pproject/ppop/image/2.mp4'
bbox = [(287, 23, 86, 320),(200,200,60,60)]
print(bbox)
while True:
    res, img = video.read()
    # k = cv.waitKey(0) & 0xFF
    # if k==ord('d'):
    #     cv.waitKey()
    #     bbox = cv.selectROI(img, False)
    timer =cv.getTickCount()

    ok,bbox=tracker.update(img)
    fps =cv.getTickFrequency()/(cv.getTickCount()-timer)