Spaces:
Runtime error
Runtime error
praysimanjuntak
commited on
Commit
•
3e68fe0
1
Parent(s):
ca16daa
Update app.py
Browse files
app.py
CHANGED
@@ -8,11 +8,9 @@ def yolov9_inference(img_path, conf_threshold, iou_threshold):
|
|
8 |
Load a YOLOv9 model, configure it, perform inference on an image, and optionally adjust
|
9 |
the input size and apply test time augmentation.
|
10 |
|
11 |
-
:param model_path: Path to the YOLOv9 model file.
|
12 |
:param conf_threshold: Confidence threshold for NMS.
|
13 |
:param iou_threshold: IoU threshold for NMS.
|
14 |
:param img_path: Path to the image file.
|
15 |
-
:param size: Optional, input size for inference.
|
16 |
:return: A tuple containing the detections (boxes, scores, categories) and the results object for further actions like displaying.
|
17 |
"""
|
18 |
# Import YOLOv9
|
@@ -80,8 +78,6 @@ def app():
|
|
80 |
fn=yolov9_inference,
|
81 |
inputs=[
|
82 |
img_path,
|
83 |
-
model_path,
|
84 |
-
image_size,
|
85 |
conf_threshold,
|
86 |
iou_threshold,
|
87 |
],
|
|
|
8 |
Load a YOLOv9 model, configure it, perform inference on an image, and optionally adjust
|
9 |
the input size and apply test time augmentation.
|
10 |
|
|
|
11 |
:param conf_threshold: Confidence threshold for NMS.
|
12 |
:param iou_threshold: IoU threshold for NMS.
|
13 |
:param img_path: Path to the image file.
|
|
|
14 |
:return: A tuple containing the detections (boxes, scores, categories) and the results object for further actions like displaying.
|
15 |
"""
|
16 |
# Import YOLOv9
|
|
|
78 |
fn=yolov9_inference,
|
79 |
inputs=[
|
80 |
img_path,
|
|
|
|
|
81 |
conf_threshold,
|
82 |
iou_threshold,
|
83 |
],
|