From 83e809bfe320f1deaff4f9c5448572c8654c6178 Mon Sep 17 00:00:00 2001 From: jiajie555 Date: Fri, 22 Nov 2024 17:59:06 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E10=E7=B1=BB=E5=95=86=E5=93=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- detect.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) mode change 100755 => 100644 detect.py diff --git a/detect.py b/detect.py old mode 100755 new mode 100644 index d567b87..f339b4e --- a/detect.py +++ b/detect.py @@ -24,7 +24,6 @@ def detect(opt, model, stride, save_img=False): # Directories save_dir = Path(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok)) # increment run (save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True) # make dir - # Initialize set_logging() device = select_device(opt.device) @@ -150,7 +149,9 @@ def detect(opt, model, stride, save_img=False): names = ["6925303773908", "6924743915848", "6920152471616", "6920005772716", "6902227018162", "6920459905012", "6972194461407", "6935284412918", "6921489033706", "6904012526494", - "6923644272159", "6924882486100", "6956511907458"] + "6923644272159", "6924882486100", "6956511907458", "6972636670213", "6974526401167", + "6914973604285", "6907476024209", "6923555210479", "6948195801276", "6976642860026", + "4893888001575", "6924187820067", "6921168509256"] targets = [] for target in pred[0]: targets.append({"Class": names[int(target[5].item())], "precision": target[4].item(), "xy1": [target[0].item(), target[1].item()],