MV_ObjectTracker.py

# https://youtu.be/1FJWXOO1SRI/

import cv2

cap = cv2.VideoCapture(0)

# tracker = cv2.legacy.TrackerMOSSE_create()
tracker = cv2.TrackerCSRT_create()

success, img = cap.read()
bbox = cv2.selectROI("Tracking",img, False)
tracker.init(img,bbox)

def drawBox(img,bbox):
    x, y, w, h = int(bbox[0]),int(bbox[1]),int(bbox[2]),int(bbox[3])
    cv2.rectangle(img,(x,y),((x+w),(y+h)), (255, 0, 255), 3, 1)
    cv2.putText(img,"Object Detected", (75,75),cv2.FONT_HERSHEY_COMPLEX, 0.7, (0,255,0), 2)
    pass

while True:
    
    timer = cv2.getTickCount()
    
    success, img = cap.read()
    
    success, bbox = tracker.update(img)
    # print(type(bbox))
    
    if success:
        drawBox(img, bbox)
    else:
        cv2.putText(img,"Object Lost", (75,75),cv2.FONT_HERSHEY_COMPLEX, 0.7, (0,0,255), 2)

    
    fps = cv2.getTickFrequency() / (cv2.getTickCount() - timer)
    cv2.putText(img,"FPS: "+str(int(fps)),(75,50),cv2.FONT_HERSHEY_COMPLEX, 0.7, (0,0,255), 2)
    cv2.imshow("Tracking", img)
  
    if cv2.waitKey(1) & 0xff == ord("q"):
        break