lichen/ieemoo-ai-zhanting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
jiajie555
2024-11-27 16:14:29 +08:00
parent 275115c7b4
commit ab7a258fee
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
detect.py
View File

@ -149,9 +149,7 @@ def detect(opt, model, stride, save_img=False):
names = ["6925303773908", "6924743915848", "6920152471616", "6920005772716", "6902227018162",
"6920459905012", "6972194461407", "6935284412918", "6921489033706", "6904012526494",
"6923644272159", "6924882486100", "6956511907458", "6972636670213", "6974526401167",
"6914973604285", "6907476024209", "6923555210479", "6948195801276", "6976642860026",
"4893888001575", "6924187820067", "6921168509256"]
"6923644272159", "6924882486100", "6956511907458"]
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()],