Hugging Face's logo

Spaces:
k20hcmus
/
FishEye8K
Running on Zero

App Files Community
3
ntsc207 commited on
Commit
61089d1
·
verified ·
1 Parent(s): b542b9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
34
  if tracking_algorithm == 'deep_sort':
35
  output_path = run_deepsort(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', draw_trails=True)
36
  elif tracking_algorithm == 'strong_sort':
37
- output_path = run_strongsort(yolo_weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='cpu', strong_sort_weights = "osnet_x0_25_msmt17.pt", hide_conf= True)
38
  else:
39
  output_path = run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True)
40
  # Assuming output_path is the path to the output file
 
34
  if tracking_algorithm == 'deep_sort':
35
  output_path = run_deepsort(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', draw_trails=True)
36
  elif tracking_algorithm == 'strong_sort':
37
+ output_path = run_strongsort(yolo_weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', strong_sort_weights = "osnet_x0_25_msmt17.pt", hide_conf= True)
38
  else:
39
  output_path = run(weights=model_id, imgsz=(image_size,image_size), conf_thres=conf_threshold, iou_thres=iou_threshold, source=input_path, device='0', hide_conf= True)
40
  # Assuming output_path is the path to the output file