Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 3051 |
-
|
| 3052 |
-
|
| 3053 |
-
|
| 3054 |
-
|
| 3055 |
-
|
| 3056 |
-
|
| 3057 |
-
|
| 3058 |
-
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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"""
|