From 98d6a9453ddaf58d1a7d4b61ba1eba2f7c076e43 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Thu, 11 Jan 2024 00:45:28 +0000 Subject: [PATCH] Auto-format by Ultralytics actions --- detect.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/detect.py b/detect.py index 35c8b17ba307..8fef3ee9f4b4 100644 --- a/detect.py +++ b/detect.py @@ -225,7 +225,9 @@ def write_to_csv(image_name, prediction, confidence): # Stream results im0 = annotator.result() - im0 = ((im0.astype(np.float32) * (1 - opt.line_alpha)) + (im0s.astype(np.float32) * opt.line_alpha)).astype(np.uint8) + im0 = ((im0.astype(np.float32) * (1 - opt.line_alpha)) + (im0s.astype(np.float32) * opt.line_alpha)).astype( + np.uint8 + ) if view_img: if platform.system() == "Linux" and p not in windows: windows.append(p) @@ -291,7 +293,9 @@ def parse_opt(): parser.add_argument("--name", default="exp", help="save results to project/name") parser.add_argument("--exist-ok", action="store_true", help="existing project/name ok, do not increment") parser.add_argument("--line-thickness", default=3, type=int, help="bounding box thickness (pixels)") - parser.add_argument("--line-alpha", default=0.5, type=float, help="bounding box transparency: 0.0:transparent - 1.0:opaque") + parser.add_argument( + "--line-alpha", default=0.5, type=float, help="bounding box transparency: 0.0:transparent - 1.0:opaque" + ) parser.add_argument("--hide-labels", default=False, action="store_true", help="hide labels") parser.add_argument("--hide-conf", default=False, action="store_true", help="hide confidences") parser.add_argument("--half", action="store_true", help="use FP16 half-precision inference")