glenn-jocher
commited on
Commit
•
bbd12c7
1
Parent(s):
fc171e2
detect.py fast bug fix
Browse files
detect.py
CHANGED
@@ -64,8 +64,7 @@ def detect(save_img=False):
|
|
64 |
pred = model(img, augment=opt.augment)[0]
|
65 |
|
66 |
# Apply NMS
|
67 |
-
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
|
68 |
-
fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
|
69 |
t2 = torch_utils.time_synchronized()
|
70 |
|
71 |
# Apply Classifier
|
|
|
64 |
pred = model(img, augment=opt.augment)[0]
|
65 |
|
66 |
# Apply NMS
|
67 |
+
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
|
|
|
68 |
t2 = torch_utils.time_synchronized()
|
69 |
|
70 |
# Apply Classifier
|