Future-Tense commited on
Commit
570e1aa
1 Parent(s): 8cd14af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
11
  URL = "https://www.youtube.com/watch?v=6NBwbKMyzEE" #URL to parse
12
 
13
  # set model parameters
14
- model.overrides['conf'] = 0.25 # NMS confidence threshold
15
  model.overrides['iou'] = 0.45 # NMS IoU threshold
16
  model.overrides['agnostic_nms'] = False # NMS class-agnostic
17
  model.overrides['max_det'] = 1000 # maximum number of detections per image
 
11
  URL = "https://www.youtube.com/watch?v=6NBwbKMyzEE" #URL to parse
12
 
13
  # set model parameters
14
+ model.overrides['conf'] = 0.50 # NMS confidence threshold
15
  model.overrides['iou'] = 0.45 # NMS IoU threshold
16
  model.overrides['agnostic_nms'] = False # NMS class-agnostic
17
  model.overrides['max_det'] = 1000 # maximum number of detections per image