raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bf

Browse Source
This commit is contained in:
Till Raab 2023-10-27 11:32:48 +02:00
parent 19a7266c17
commit 0befddab8d
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
inference.py
View File

@ -69,10 +69,9 @@ def main(args):
inference_loader = create_inference_loader(inference_data) inference_loader = create_inference_loader(inference_data)
dataset_name = Path(args.folder).name dataset_name = Path(args.folder).name
embed()
quit() if not (Path(INFERENCE_OUTDIR)/dataset_name).exists():
if not (Path(OUTDIR)/dataset_name).exists(): Path(Path(INFERENCE_OUTDIR)/dataset_name).mkdir(parents=True, exist_ok=True)
Path(Path(OUTDIR)/dataset_name).mkdir(parents=True, exist_ok=True)
infere_model(inference_loader, model, dataset_name) infere_model(inference_loader, model, dataset_name)