Spaces:
Sleeping
Sleeping
Commit ·
5390360
1
Parent(s): de3e659
Added smaller model and changed examples
Browse files- models/{best.pt → yolo11m.pt} +0 -0
- models/yolo11s.pt +3 -0
- yolo-inference-project/src/app.py +5 -5
models/{best.pt → yolo11m.pt}
RENAMED
|
File without changes
|
models/yolo11s.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:540894613f3563a4537865c39131125003c0cccde281cd4bb16fc86ce36f4b2c
|
| 3 |
+
size 19222483
|
yolo-inference-project/src/app.py
CHANGED
|
@@ -20,10 +20,10 @@ def get_model_names():
|
|
| 20 |
|
| 21 |
model_names = get_model_names()
|
| 22 |
|
| 23 |
-
image_paths= [['examples/smaller_many_cans.jpg', '
|
| 24 |
-
['examples/gazebo_all.jpg', '
|
| 25 |
-
['examples/sequence_unity.jpg', '
|
| 26 |
-
['examples/real_plank.jpg', '
|
| 27 |
]
|
| 28 |
|
| 29 |
demo = gr.Interface(
|
|
@@ -37,7 +37,7 @@ demo = gr.Interface(
|
|
| 37 |
title="YOLO Model Inference",
|
| 38 |
description="Select a YOLO model, upload an image, and set the confidence threshold to perform inference.",
|
| 39 |
examples=image_paths,
|
| 40 |
-
|
| 41 |
)
|
| 42 |
|
| 43 |
demo.launch()
|
|
|
|
| 20 |
|
| 21 |
model_names = get_model_names()
|
| 22 |
|
| 23 |
+
image_paths= [['examples/smaller_many_cans.jpg', 'yolo11m.pt', 0.5],
|
| 24 |
+
['examples/gazebo_all.jpg', 'yolo11m.pt', 0.5],
|
| 25 |
+
['examples/sequence_unity.jpg', 'yolo11m.pt', 0.5],
|
| 26 |
+
['examples/real_plank.jpg', 'yolo11m.pt', 0.5],
|
| 27 |
]
|
| 28 |
|
| 29 |
demo = gr.Interface(
|
|
|
|
| 37 |
title="YOLO Model Inference",
|
| 38 |
description="Select a YOLO model, upload an image, and set the confidence threshold to perform inference.",
|
| 39 |
examples=image_paths,
|
| 40 |
+
# flagging_mode="auto"
|
| 41 |
)
|
| 42 |
|
| 43 |
demo.launch()
|