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

embed for bf

Browse Source
This commit is contained in:
Till Raab 2023-10-27 11:22:59 +02:00
parent 71030bb707
commit 873758d461
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
datasets.py
View File

@ -24,7 +24,8 @@ class InferenceDataset(Dataset):
def __init__(self, dir_path): def __init__(self, dir_path):
self.dir_path = dir_path self.dir_path = dir_path
self.all_images = sorted(list(Path(self.dir_path).rglob(f'*.png'))) self.all_images = sorted(list(Path(self.dir_path).rglob(f'*.png')))
embed()
quit()
def __len__(self): def __len__(self):
return len(self.all_images) return len(self.all_images)

2
inference.py
View File

@ -69,8 +69,6 @@ def main(args):
inference_data = InferenceDataset(args.folder) inference_data = InferenceDataset(args.folder)
inference_loader = create_inference_loader(inference_data) inference_loader = create_inference_loader(inference_data)
embed()
quit()
dataset_name = Path(args.folder).name dataset_name = Path(args.folder).name
infere_model(inference_loader, model, dataset_name) infere_model(inference_loader, model, dataset_name)