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

新增10类商品

Browse Source
This commit is contained in:
jiajie555
2024-11-22 17:59:06 +08:00
parent a95db2a8fb
commit 83e809bfe3
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
detect.py Executable file → Normal file
View File

@ -24,7 +24,6 @@ def detect(opt, model, stride, save_img=False):
# Directories # Directories
save_dir = Path(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok)) # increment run 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 (save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True) # make dir
# Initialize # Initialize
set_logging() set_logging()
device = select_device(opt.device) device = select_device(opt.device)
@ -150,7 +149,9 @@ def detect(opt, model, stride, save_img=False):
names = ["6925303773908", "6924743915848", "6920152471616", "6920005772716", "6902227018162", names = ["6925303773908", "6924743915848", "6920152471616", "6920005772716", "6902227018162",
"6920459905012", "6972194461407", "6935284412918", "6921489033706", "6904012526494", "6920459905012", "6972194461407", "6935284412918", "6921489033706", "6904012526494",
"6923644272159", "6924882486100", "6956511907458"] "6923644272159", "6924882486100", "6956511907458", "6972636670213", "6974526401167",
"6914973604285", "6907476024209", "6923555210479", "6948195801276", "6976642860026",
"4893888001575", "6924187820067", "6921168509256"]
targets = [] targets = []
for target in pred[0]: for target in pred[0]:
targets.append({"Class": names[int(target[5].item())], "precision": target[4].item(), "xy1": [target[0].item(), target[1].item()], targets.append({"Class": names[int(target[5].item())], "precision": target[4].item(), "xy1": [target[0].item(), target[1].item()],