Spaces:

linhcuem
/

cham_diem_yolov8

Sleeping

linhcuem commited on Oct 24, 2023

Commit

151e108

1 Parent(s): f07963f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,12 +38,12 @@ def yolov8_img_inference(
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
     model = YOLO(model_path)
-    # model.conf = conf_threshold
-    # model.iou = iou_threshold
-    model.overrides['conf'] = conf_threshold
-    model.overrides['iou'] = iou_threshold
-    model.overrides['agnostic_nms'] = False
-    model.overrides['max_det'] = 1000
     # image = read_image
     results = model.predict(image)
     render = render_result(model=model, image=image, result=results[0])

     iou_threshold: gr.inputs.Slider = 0.45,
 ):
     model = YOLO(model_path)
+    model.conf = conf_threshold
+    model.iou = iou_threshold
+    # model.overrides['conf'] = conf_threshold
+    # model.overrides['iou'] = iou_threshold
+    # model.overrides['agnostic_nms'] = False
+    # model.overrides['max_det'] = 1000
     # image = read_image
     results = model.predict(image)
     render = render_result(model=model, image=image, result=results[0])