hamhanry commited on
Commit
c9b6fea
·
1 Parent(s): bb69e8f

add s,b,m,l,x pretrained model

Browse files
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