Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,19 +9,12 @@ import numpy as np
|
|
9 |
import uuid
|
10 |
import spaces
|
11 |
|
12 |
-
|
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]}
|
|
|
25 |
print(modin)
|
26 |
model = YOLO(modin)
|
27 |
|
|
|
9 |
import uuid
|
10 |
import spaces
|
11 |
|
12 |
+
ver=[6,7,8.9,10,11]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
ltr=["n","s","m","1","x"]
|
14 |
+
tsk=["","-seg","-pose","-obb","-cls"]
|
15 |
#yolov8s.pt
|
16 |
+
modin=f"yolov{ver[2]}{ltr[1]}{tsk[0]}.pt"
|
17 |
+
if
|
18 |
print(modin)
|
19 |
model = YOLO(modin)
|
20 |
|