This commit is contained in:
huangtao
2022-07-27 17:55:52 +08:00
parent 7edddd1378
commit 4a15c93caf
5 changed files with 10 additions and 10 deletions

View File

@ -143,7 +143,7 @@ def detect(opt, save_img=False):
s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
print(f"Results saved to {save_dir}{s}")
print(f'Done. ({time.time() - t0:.3f}s)')
# print(f'Done. ({time.time() - t0:.3f}s)')
names = ["6925303773908", "6924743915848", "6920152471616", "6920005772716", "6902227018162",
"6920459905012", "6972194461407", "6935284412918", "6921489033706", "6904012526494",
@ -153,7 +153,6 @@ def detect(opt, save_img=False):
targets.append({"Class": names[int(target[5].item())], "precision": target[4].item(), "xy1": [target[0].item(), target[1].item()],
"xy2": [target[2].item(), target[3].item()]})
resu = {"TargetDetect": targets}
print(resu)
return resu