提交 bbd12c7c authored 作者: Glenn Jocher's avatar Glenn Jocher

detect.py fast bug fix

上级 fc171e26
...@@ -64,8 +64,7 @@ def detect(save_img=False): ...@@ -64,8 +64,7 @@ def detect(save_img=False):
pred = model(img, augment=opt.augment)[0] pred = model(img, augment=opt.augment)[0]
# Apply NMS # Apply NMS
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
t2 = torch_utils.time_synchronized() t2 = torch_utils.time_synchronized()
# Apply Classifier # Apply Classifier
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论