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

5 years ago · ad71d2d513
parent 8f17a96ff0
commit ad71d2d513
1 changed files with 1 additions and 1 deletions
--- 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: