validation/create_ground_truth.py

@@ -1,30 +1,58 @@
 import cv2
 import sys
  
-tracker = cv2.TrackerMIL_create()
+(major_ver, minor_ver, subminor_ver) = (cv2.__version__).split('.')
  
-# Read video
+if __name__ == '__main__' :
-video = cv2.VideoCapture("./validation/TestVideo.mp4")
  
-# Exit if video not opened.
+    # Set up tracker.
-if not video.isOpened():
+    # 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()
+        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
+    video = cv2.VideoCapture("./validation/TestVideo.mp4")
+ 
+    # Exit if video not opened.
+    if not video.isOpened():
         print("Could not open video")
         sys.exit()
  
-# Read first frame.
+    # Read first frame.
-ok, frame = video.read()
+    ok, frame = video.read()
-if not ok:
+    if not ok:
         print('Cannot read video file')
         sys.exit()
      
-# Define an initial bounding box
+    # Define an initial bounding box
-bbox = (857, 189, 346, 434)
+    bbox = (857, 189, 346, 434)
  
-# Initialize tracker with first frame and bounding box
+    # Initialize tracker with first frame and bounding box
-ok = tracker.init(frame, bbox)
+    ok = tracker.init(frame, bbox)
  
-frame_count = 0
+    frame_count = 0
-while True:
+    while True:
         # Read a new frame
         ok, frame = video.read()
         if not ok:
@@ -53,12 +81,16 @@ while True:
             # Tracking failure
             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
-    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
         cv2.imshow("Tracking", frame)
 
+
         frame_count += 1
  
         # Exit if ESC pressed