Spaces:
Sleeping
Sleeping
cv
Browse files
app.py
CHANGED
|
@@ -138,6 +138,17 @@ with gr.Blocks() as app:
|
|
| 138 |
value="Image",
|
| 139 |
label="Input Type",
|
| 140 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 141 |
conf_threshold = gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence Threshold")
|
| 142 |
iou_threshold = gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU Threshold")
|
| 143 |
max_detection = gr.Slider(minimum=1, maximum=300, step=1, value=300, label="Max Detection")
|
|
|
|
| 138 |
value="Image",
|
| 139 |
label="Input Type",
|
| 140 |
)
|
| 141 |
+
model_id = gr.Dropdown(
|
| 142 |
+
label="Model Name",
|
| 143 |
+
choices=[
|
| 144 |
+
'yolo11n.pt', 'yolo11s.pt', 'yolo11m.pt', 'yolo11l.pt', 'yolo11x.pt',
|
| 145 |
+
'yolo11n-seg.pt', 'yolo11s-seg.pt', 'yolo11m-seg.pt', 'yolo11l-seg.pt', 'yolo11x-seg.pt',
|
| 146 |
+
'yolo11n-pose.pt', 'yolo11s-pose.pt', 'yolo11m-pose.pt', 'yolo11l-pose.pt', 'yolo11x-pose.pt',
|
| 147 |
+
'yolo11n-obb.pt', 'yolo11s-obb.pt', 'yolo11m-obb.pt', 'yolo11l-obb.pt', 'yolo11x-obb.pt',
|
| 148 |
+
'yolo11n-cls.pt', 'yolo11s-cls.pt', 'yolo11m-cls.pt', 'yolo11l-cls.pt', 'yolo11x-cls.pt'
|
| 149 |
+
],
|
| 150 |
+
value="yolo11n.pt",
|
| 151 |
+
)
|
| 152 |
conf_threshold = gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence Threshold")
|
| 153 |
iou_threshold = gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU Threshold")
|
| 154 |
max_detection = gr.Slider(minimum=1, maximum=300, step=1, value=300, label="Max Detection")
|