glenn-jocher commited on
Commit
ad71d2d
1 Parent(s): 8f17a96

update times to end-to-end latency (inference + NMS)

Browse files
Files changed (1) hide show
  1. detect.py +1 -1
detect.py CHANGED
@@ -62,7 +62,6 @@ def detect(save_img=False):
62
  # Inference
63
  t1 = torch_utils.time_synchronized()
64
  pred = model(img, augment=opt.augment)[0]
65
- t2 = torch_utils.time_synchronized()
66
 
67
  # to float
68
  if half:
@@ -71,6 +70,7 @@ def detect(save_img=False):
71
  # Apply NMS
72
  pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
73
  fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
 
74
 
75
  # Apply Classifier
76
  if classify:
 
62
  # Inference
63
  t1 = torch_utils.time_synchronized()
64
  pred = model(img, augment=opt.augment)[0]
 
65
 
66
  # to float
67
  if half:
 
70
  # Apply NMS
71
  pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
72
  fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
73
+ t2 = torch_utils.time_synchronized()
74
 
75
  # Apply Classifier
76
  if classify: