diff --git a/detect.py b/detect.py index bf8d2e4d..7be56115 100644 --- a/detect.py +++ b/detect.py @@ -88,7 +88,8 @@ def detect(save_img=False): t2 = torch_utils.time_synchronized() # Apply NMS - pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms) + pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, + multi_label=False, classes=opt.classes, agnostic=opt.agnostic_nms) # Apply Classifier if classify: diff --git a/utils/utils.py b/utils/utils.py index e5a4c526..c5da3a27 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -828,7 +828,7 @@ def fitness(x): # Plotting functions --------------------------------------------------------------------------------------------------- def plot_one_box(x, img, color=None, label=None, line_thickness=None): # Plots one bounding box on image img - tl = line_thickness or round(0.002 * (img.shape[0] + img.shape[1]) / 2) + 1 # line thickness + tl = line_thickness or max(round(0.002 * (img.shape[0] + img.shape[1]) / 2), 1) # line thickness color = color or [random.randint(0, 255) for _ in range(3)] c1, c2 = (int(x[0]), int(x[1])), (int(x[2]), int(x[3])) cv2.rectangle(img, c1, c2, color, thickness=tl)