|
@@ -91,7 +91,7 @@ def detect(save_img=False):
|
|
|
# Apply NMS
|
|
|
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
|
|
|
t3 = time_synchronized()
|
|
|
-
|
|
|
+
|
|
|
# Apply Classifier
|
|
|
if classify:
|
|
|
pred = apply_classifier(pred, modelc, img, im0s)
|
|
@@ -126,7 +126,7 @@ def detect(save_img=False):
|
|
|
|
|
|
if save_img or view_img: # Add bbox to image
|
|
|
label = f'{names[int(cls)]} {conf:.2f}'
|
|
|
- plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=1)
|
|
|
+ plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thinckness=1)
|
|
|
|
|
|
# Print time (inference + NMS)
|
|
|
print(f'{s}Done. ({(1E3 * (t2 - t1)):.1f}ms) Inference, ({(1E3 * (t3 - t2)):.1f}ms) NMS')
|