From ad71d2d513dd1cce71b87b72af6c2685709549ad Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 12 Jun 2020 11:42:08 -0700
Subject: [PATCH] update times to end-to-end latency (inference + NMS)

---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index b60d283..132d162 100644
--- a/detect.py
+++ b/detect.py
@@ -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: