OpenCV Python, однообъектный трекер. После выбора области интереса я получаю сообщение об ошибке «AttributeError: у объекта« NoneType »нет атрибута« init »» - PullRequest
0 голосов
/ 29 апреля 2020
import cv2

def choose_tracker():
    print("-----------Tracker Menu-----------\n1.Boosting\n2.MIL\n3.KCF\n4.TLD\n5.Median Flow\n6.Goturn\n7.MOSSE\n8.CSRT")
    choice = input("Select any of the above tracker: ")

    if choice == 1:
        tracker = cv2.TrackerBoosting_create()
    if choice == 2:
        tracker = cv2.TrackerMIL_create()
    if choice == 3:
        tracker = cv2.TrackerKCF_create()
    if choice == 4:
        tracker = cv2.TrackerTLD_create()
    if choice == 5:
        tracker = cv2.TrackerMedianFlow_create()
    if choice == 6:
        tracker = cv2.TrackerGOTURN_create()
    if choice == 7:
        tracker = cv2.TrackerMOSSE_create()
    if choice == 8:
        tracker = cv2.TrackerCSRT_create()    

#call the choose_tracker function
tracker = choose_tracker()

#get tracker name
tracker_name = str(tracker).split()[0][1:]

#get video
cap = cv2.VideoCapture('Vehicles.mp4')

ret, frame = cap.read()

# Define an initial bounding bo
bbox = (287, 23, 86, 320)

bbox = cv2.selectROI(frame,False)

#Initializing the tracker
ret = tracker.init(frame,bbox)

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

    #update the frame by tracker  
    success,roi = tracker.update(frame)

    #convert roi from tuple to int
    (x,y,w,h) = tuple(map(int,roi))

    if success:
        pts1 = (x,y)
        pts2 = (x+w,y+h)
        cv2.rectangle(frame,pts1,pts2,(255,0,0),2)
    else:
        cv2.putText(frame,"Failed to Track",(100,200),cv2.FONT_HERSHEY_SIMPLEX,(255,255,0),3)

    #Display Tracker
    cv2.putText(frame,tracker_name,(20,400),cv2.FONT_HERSHEY_SIMPLEX,1,(255,0,255),3)
    cv2.imshow(tracker_name,frame)

    if cv2.waitKey(50) & 0xFF == ord("q"):
        break

cap.release()
cv2.destroyAllWindows()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...