CosmickVisions commited on
Commit
2222f26
·
verified ·
1 Parent(s): 5ca5bfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -16
app.py CHANGED
@@ -3047,23 +3047,22 @@ def stabilize_frame(frame, prev_frame_gray, current_frame_gray):
3047
  return stabilized_frame
3048
 
3049
  def create_tracker(tracker_type="CSRT"):
3050
- """Create an OpenCV tracker of the specified type"""
3051
- if tracker_type == 'BOOSTING':
3052
- return cv2.legacy.TrackerBoosting_create()
3053
- elif tracker_type == 'MIL':
3054
- return cv2.legacy.TrackerMIL_create()
3055
- elif tracker_type == 'KCF':
3056
- return cv2.legacy.TrackerKCF_create()
3057
- elif tracker_type == 'TLD':
3058
- return cv2.legacy.TrackerTLD_create()
3059
- elif tracker_type == 'MEDIANFLOW':
3060
- return cv2.legacy.TrackerMedianFlow_create()
3061
- elif tracker_type == 'CSRT':
3062
- return cv2.legacy.TrackerCSRT_create()
3063
- elif tracker_type == 'MOSSE':
3064
- return cv2.legacy.TrackerMOSSE_create()
3065
  else:
3066
- return cv2.legacy.TrackerCSRT_create() # Default
 
 
 
 
 
3067
 
3068
  class ObjectTracker:
3069
  """Manages object tracking across video frames"""
 
3047
  return stabilized_frame
3048
 
3049
  def create_tracker(tracker_type="CSRT"):
3050
+ (major_ver, minor_ver, subminor_ver) = cv2.__version__.split('.')
3051
+
3052
+ # For OpenCV 4.5.x and newer
3053
+ if int(major_ver) >= 4 and int(minor_ver) >= 5:
3054
+ if tracker_type == "CSRT":
3055
+ return cv2.legacy.TrackerCSRT_create()
3056
+ elif tracker_type == "KCF":
3057
+ return cv2.legacy.TrackerKCF_create()
3058
+ # Add other tracker types as needed
 
 
 
 
 
 
3059
  else:
3060
+ # For older OpenCV versions
3061
+ if tracker_type == "CSRT":
3062
+ return cv2.TrackerCSRT_create()
3063
+ elif tracker_type == "KCF":
3064
+ return cv2.TrackerKCF_create()
3065
+ # Add other tracker types as needed
3066
 
3067
  class ObjectTracker:
3068
  """Manages object tracking across video frames"""