Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,11 +33,11 @@ model = YOLO('linhcuem/cham_diem_yolov8')
|
|
33 |
def yolov8_img_inference(
|
34 |
image: gr.inputs.Image = None,
|
35 |
model_path: gr.inputs.Dropdown = None,
|
36 |
-
image_size
|
37 |
-
conf_threshold
|
38 |
-
iou_threshold
|
39 |
):
|
40 |
-
model = YOLO(model_path)
|
41 |
model.conf = conf_threshold
|
42 |
model.iou = iou_threshold
|
43 |
# model.overrides['conf'] = conf_threshold
|
|
|
33 |
def yolov8_img_inference(
|
34 |
image: gr.inputs.Image = None,
|
35 |
model_path: gr.inputs.Dropdown = None,
|
36 |
+
image_size = 640,
|
37 |
+
conf_threshold = 0.25,
|
38 |
+
iou_threshold = 0.45,
|
39 |
):
|
40 |
+
# model = YOLO(model_path)
|
41 |
model.conf = conf_threshold
|
42 |
model.iou = iou_threshold
|
43 |
# model.overrides['conf'] = conf_threshold
|