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()],