Glenn Jocher
4 years ago
1 changed files with 1 additions and 2 deletions
+ 1
- 2
detect.py
View File
| 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, | |||||
| fast=True, classes=opt.classes, agnostic=opt.agnostic_nms) | |||||
| pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms) | |||||
| t2 = torch_utils.time_synchronized() | t2 = torch_utils.time_synchronized() | ||||
| # Apply Classifier | # Apply Classifier |
Loading…