praysimanjuntak commited on
Commit
3e68fe0
1 Parent(s): ca16daa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
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
  ],