Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,22 @@ import gradio as gr
|
|
2 |
import cv2
|
3 |
import numpy as np
|
4 |
from ultralytics import YOLO
|
5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
# Load YOLOv8 model (pre-trained on COCO dataset)
|
8 |
model = YOLO("yolov8x.pt") # Highest accuracy version
|
|
|
2 |
import cv2
|
3 |
import numpy as np
|
4 |
from ultralytics import YOLO
|
5 |
+
|
6 |
+
import os
|
7 |
+
import subprocess
|
8 |
+
|
9 |
+
# Clone the SORT repository
|
10 |
+
if not os.path.isdir("sort"):
|
11 |
+
subprocess.run(["git", "clone", "https://github.com/abewley/sort.git"])
|
12 |
+
|
13 |
+
# Add sort module to the Python path
|
14 |
+
import sys
|
15 |
+
sys.path.append(os.path.join(os.getcwd(), "sort"))
|
16 |
+
|
17 |
+
from sort import Sort
|
18 |
+
|
19 |
+
|
20 |
+
#from sort import Sort # SORT tracker
|
21 |
|
22 |
# Load YOLOv8 model (pre-trained on COCO dataset)
|
23 |
model = YOLO("yolov8x.pt") # Highest accuracy version
|