Spaces:
Running
Running
add s,b,m,l,x pretrained model
Browse files- app.py +15 -0
- pretrained/yolov10b-obb.pt +3 -0
- pretrained/yolov10l-obb.pt +3 -0
- pretrained/yolov10m-obb.pt +3 -0
- pretrained/yolov10s-obb.pt +3 -0
- pretrained/yolov10x-obb.pt +3 -0
app.py
CHANGED
@@ -9,6 +9,16 @@ def inference(image, video, model_id, image_size, conf_threshold):
|
|
9 |
model = YOLO("pretrained/yolov10n-obb.pt")
|
10 |
elif model_id == "yolov10s-640-obb":
|
11 |
model = YOLO("pretrained/yolov10s-640-obb.pt")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
|
13 |
if image:
|
14 |
results = model.predict(source=image, imgsz=image_size, conf=conf_threshold, device="cpu")
|
@@ -62,6 +72,11 @@ def app():
|
|
62 |
choices=[
|
63 |
"yolov10n-obb",
|
64 |
"yolov10s-640-obb",
|
|
|
|
|
|
|
|
|
|
|
65 |
],
|
66 |
value="yolov10n-obb",
|
67 |
)
|
|
|
9 |
model = YOLO("pretrained/yolov10n-obb.pt")
|
10 |
elif model_id == "yolov10s-640-obb":
|
11 |
model = YOLO("pretrained/yolov10s-640-obb.pt")
|
12 |
+
elif model_id == "yolov10s-obb":
|
13 |
+
model = YOLO("pretrained/yolov10s-obb.pt")
|
14 |
+
elif model_id == "yolov10m-obb":
|
15 |
+
model = YOLO("pretrained/yolov10m-obb.pt")
|
16 |
+
elif model_id == "yolov10b-obb":
|
17 |
+
model = YOLO("pretrained/yolov10b-obb.pt")
|
18 |
+
elif model_id == "yolov10l-obb":
|
19 |
+
model = YOLO("pretrained/yolov10l-obb.pt")
|
20 |
+
elif model_id == "yolov10x-obb":
|
21 |
+
model = YOLO("pretrained/yolov10x-obb.pt")
|
22 |
|
23 |
if image:
|
24 |
results = model.predict(source=image, imgsz=image_size, conf=conf_threshold, device="cpu")
|
|
|
72 |
choices=[
|
73 |
"yolov10n-obb",
|
74 |
"yolov10s-640-obb",
|
75 |
+
"yolov10s-obb",
|
76 |
+
"yolov10m-obb",
|
77 |
+
"yolov10b-obb",
|
78 |
+
"yolov10l-obb",
|
79 |
+
"yolov10x-obb",
|
80 |
],
|
81 |
value="yolov10n-obb",
|
82 |
)
|
pretrained/yolov10b-obb.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd0ccd2634abfbb25e2a8c184605649e10fd37e3cf5d019c27ef30358dfbffef
|
3 |
+
size 48595832
|
pretrained/yolov10l-obb.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cf73ff8041d7abf353d0f095b2e1a4714f21dd9c60c8d9d942f63e3a78b5fd34
|
3 |
+
size 59370808
|
pretrained/yolov10m-obb.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f21a6e383e866a0d0566f75343bec6021999c27449b926b8f7308cfd2848f945
|
3 |
+
size 5646504
|
pretrained/yolov10s-obb.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d5c2ff9a6f20ab31d1ebccbd96971f5fad99a6f58efd93d8ac3281d544790af
|
3 |
+
size 18019120
|
pretrained/yolov10x-obb.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:baf306dc88346afd145f37bb336befb2ec870654d08ed4d47fd81e6d88d05fcb
|
3 |
+
size 75384310
|