Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,22 @@ import time
|
|
| 8 |
import numpy as np
|
| 9 |
import uuid
|
| 10 |
import spaces
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
@spaces.GPU
|
| 14 |
def stream_object_detection(video):
|
|
|
|
| 8 |
import numpy as np
|
| 9 |
import uuid
|
| 10 |
import spaces
|
| 11 |
+
|
| 12 |
+
print(YOLO)
|
| 13 |
+
|
| 14 |
+
"""| YOLOv8 | `yolov8n.pt` `yolov8s.pt` `yolov8m.pt` `yolov8l.pt` `yolov8x.pt` …
|
| 15 |
+
| YOLOv8-seg | `yolov8n-seg.pt` `yolov8s-seg.pt` `yolov8m-seg.pt` `yolov8l-seg.pt` `yolov8x-seg.pt` …
|
| 16 |
+
| YOLOv8-pose | `yolov8n-pose.pt` `yolov8s-pose.pt` `yolov8m-pose.pt` `yolov8l-pose.pt` `yolov8x-pose.pt` `yolov8x-pose…
|
| 17 |
+
| YOLOv8-obb | `yolov8n-obb.pt` `yolov8s-obb.pt` `yolov8m-obb.pt` `yolov8l-obb.pt` `yolov8x-obb.pt` …
|
| 18 |
+
| YOLOv8-cls | `yolov8n-cls.pt` `yolov8s-cls.pt` `yolov8m-cls.pt` `yolov8l-cls.pt` `yolov8x-cls.pt` """
|
| 19 |
+
|
| 20 |
+
ver=[6:11]
|
| 21 |
+
ltr=["n","s","m","1","x"]
|
| 22 |
+
tsk=["","seg","pose","obb","cls"]
|
| 23 |
+
#yolov8s.pt
|
| 24 |
+
modin=f"yolov{ver[2]{ltr[1]}-{tsk[0]}.pt"
|
| 25 |
+
print(modin)
|
| 26 |
+
model = YOLO(modin)
|
| 27 |
|
| 28 |
@spaces.GPU
|
| 29 |
def stream_object_detection(video):
|