Spaces:
Runtime error
Runtime error
Future-Tense
commited on
Commit
·
570e1aa
1
Parent(s):
8cd14af
Update app.py
Browse files
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.
|
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
|