使用python中的边界框更新OpenCV跟踪器

Sha*_*lgi 5 python opencv

我正在使用OpenCV跟踪器在视频中执行人脸跟踪,并每隔几帧就使用一个人脸检测器。如果面部检测器检测到面部,我想使用“检测到的”边界框更新跟踪器。我看到有一个选项可以在C ++实现中输入Rect,但是由于某种原因,不能在opencv文档中编写的python实现中输入。使用dlib的correlation_tracker时,这也是一个选项。

目前,我只能用边界框初始化跟踪器,而不能在Python中用边界框更新跟踪器。如果我的跟踪器已经偏离了要跟踪的初始面部,即使知道了面部现在在哪里,也无法“带回”(使用面部检测器)。有没有办法在python中做到这一点(例如,我应该杀死当前的跟踪器并使用检测到的边界框初始化另一个跟踪器)?

小智 6

我一直在寻找相同的东西,每次成功检测时,我都通过重新创建跟踪器找到了解决问题的方法。请检查以下代码。如果有什么不清楚的,请随时询问详细信息:

import cv2 as cv

cap = cv.VideoCapture(0)

face_front_cascade = cv.CascadeClassifier("haarcascade_frontalface_alt.xml")    
tracker = cv.TrackerKCF_create()
bbox = ()

while True:
    ret,frame = cap.read()

    #press S to capture the face
    if cv.waitKey(20) & 0xFF == ord("s"):
        frame_gray = cv.cvtColor(frame,cv.COLOR_BGR2GRAY)
        face = face_front_cascade.detectMultiScale(frame_gray, scaleFactor=1.5, minNeighbors=3)
        for (x, y, w, h) in face: 
            colour = (0,0,255)
            stroke = 20
            cv.rectangle(frame,(x,y),(x+w,y+h), colour, stroke)
            bbox = (x,y,w,h)
            tracker = cv.TrackerKCF_create() #overwrite old tracker

    #trace face and draw box around it
    if bbox:
        tracker.init(frame, bbox)
        ret, bbox = tracker.update(frame)
        if ret:
            p1 = (int(bbox[0]), int(bbox[1]))
            p2 = (int(bbox[0] + bbox[2]), int(bbox[1] + bbox[3]))
            cv.rectangle(frame, p1, p2, (255, 0, 0), 2, 1)

    #show result
    cv.imshow("frame",frame)

    #press ESC to exit
    if cv.waitKey(20) & 0xFF ==27:
        break    
cap.release()
cv.destroyAllWindows()
Run Code Online (Sandbox Code Playgroud)