Commit
·
ad71d2d
1
Parent(s):
8f17a96
update times to end-to-end latency (inference + NMS)
Browse files
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:
|