Compare commits
2 commits
d040135d02
...
fd79b30f52
Author | SHA1 | Date | |
---|---|---|---|
|
fd79b30f52 | ||
|
7ddd813361 |
1 changed files with 57 additions and 89 deletions
|
@ -1,35 +1,7 @@
|
||||||
import cv2
|
import cv2
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
(major_ver, minor_ver, subminor_ver) = (cv2.__version__).split('.')
|
|
||||||
|
|
||||||
if __name__ == '__main__' :
|
|
||||||
|
|
||||||
# Set up tracker.
|
|
||||||
# Instead of MIL, you can also use
|
|
||||||
|
|
||||||
tracker_types = ['BOOSTING', 'MIL','KCF', 'TLD', 'MEDIANFLOW', 'GOTURN', 'MOSSE', 'CSRT']
|
|
||||||
tracker_type = tracker_types[1]
|
|
||||||
|
|
||||||
if int(minor_ver) < 3:
|
|
||||||
tracker = cv2.Tracker_create(tracker_type)
|
|
||||||
else:
|
|
||||||
if tracker_type == 'BOOSTING':
|
|
||||||
tracker = cv2.TrackerBoosting_create()
|
|
||||||
if tracker_type == 'MIL':
|
|
||||||
tracker = cv2.TrackerMIL_create()
|
tracker = cv2.TrackerMIL_create()
|
||||||
if tracker_type == 'KCF':
|
|
||||||
tracker = cv2.TrackerKCF_create()
|
|
||||||
if tracker_type == 'TLD':
|
|
||||||
tracker = cv2.TrackerTLD_create()
|
|
||||||
if tracker_type == 'MEDIANFLOW':
|
|
||||||
tracker = cv2.TrackerMedianFlow_create()
|
|
||||||
if tracker_type == 'GOTURN':
|
|
||||||
tracker = cv2.TrackerGOTURN_create()
|
|
||||||
if tracker_type == 'MOSSE':
|
|
||||||
tracker = cv2.TrackerMOSSE_create()
|
|
||||||
if tracker_type == "CSRT":
|
|
||||||
tracker = cv2.TrackerCSRT_create()
|
|
||||||
|
|
||||||
# Read video
|
# Read video
|
||||||
video = cv2.VideoCapture("./validation/TestVideo.mp4")
|
video = cv2.VideoCapture("./validation/TestVideo.mp4")
|
||||||
|
@ -81,16 +53,12 @@ if __name__ == '__main__' :
|
||||||
# Tracking failure
|
# Tracking failure
|
||||||
cv2.putText(frame, "Tracking failure detected", (100,80), cv2.FONT_HERSHEY_SIMPLEX, 0.75,(0,0,255),2)
|
cv2.putText(frame, "Tracking failure detected", (100,80), cv2.FONT_HERSHEY_SIMPLEX, 0.75,(0,0,255),2)
|
||||||
|
|
||||||
# Display tracker type on frame
|
|
||||||
cv2.putText(frame, tracker_type + " Tracker", (100,20), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50,170,50),2);
|
|
||||||
|
|
||||||
# Display FPS on frame
|
# Display FPS on frame
|
||||||
cv2.putText(frame, "FPS : " + str(int(fps)), (100,50), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50,170,50), 2);
|
cv2.putText(frame, "FPS : " + str(int(fps)), (100,50), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (50,170,50), 2)
|
||||||
|
|
||||||
# Display result
|
# Display result
|
||||||
cv2.imshow("Tracking", frame)
|
cv2.imshow("Tracking", frame)
|
||||||
|
|
||||||
|
|
||||||
frame_count += 1
|
frame_count += 1
|
||||||
|
|
||||||
# Exit if ESC pressed
|
# Exit if ESC pressed
|
||||||
|
|
Loading…
Reference in a new issue