@@ -62,7 +62,6 @@ def detect(save_img=False): | |||
# Inference | |||
t1 = torch_utils.time_synchronized() | |||
pred = model(img, augment=opt.augment)[0] | |||
t2 = torch_utils.time_synchronized() | |||
# to float | |||
if half: | |||
@@ -71,6 +70,7 @@ def detect(save_img=False): | |||
# Apply NMS | |||
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, | |||
fast=True, classes=opt.classes, agnostic=opt.agnostic_nms) | |||
t2 = torch_utils.time_synchronized() | |||
# Apply Classifier | |||
if classify: |