From 26ed5e2ddcf3b2a0324a25f1d2e5994f3ed447e4 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 23 Dec 2019 11:25:15 -0800 Subject: [PATCH] updates --- detect.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/detect.py b/detect.py index 051ca6fc..034a1184 100644 --- a/detect.py +++ b/detect.py @@ -88,7 +88,7 @@ def detect(save_txt=False, save_img=False): # Apply NMS pred = non_max_suppression(pred, opt.conf_thres, opt.nms_thres) - # Apply + # Apply Classifier if classify: pred = apply_classifier(pred, modelc, img, im0s) @@ -105,6 +105,9 @@ def detect(save_txt=False, save_img=False): # Rescale boxes from img_size to im0 size det[:, :4] = scale_coords(img.shape[2:], det[:, :4], im0.shape).round() + # Print time (inference + NMS) + print('%sDone. (%.3fs)' % (s, time.time() - t)) + # Print results for c in det[:, -1].unique(): n = (det[:, -1] == c).sum() # detections per class @@ -120,8 +123,6 @@ def detect(save_txt=False, save_img=False): label = '%s %.2f' % (names[int(cls)], conf) plot_one_box(xyxy, im0, label=label, color=colors[int(cls)]) - print('%sDone. (%.3fs)' % (s, time.time() - t)) - # Stream results if view_img: cv2.imshow(p, im0)