embed for bf

2023-10-27 11:22:59 +02:00 · 2023-10-27 11:22:59 +02:00 · 873758d461
commit 873758d461
parent 71030bb707
2 changed files with 2 additions and 3 deletions
--- a/datasets.py
+++ b/datasets.py
@ -24,7 +24,8 @@ class InferenceDataset(Dataset):
    def __init__(self, dir_path):
        self.dir_path = dir_path
        self.all_images = sorted(list(Path(self.dir_path).rglob(f'*.png')))
-
+        embed()
+        quit()
    def __len__(self):
        return len(self.all_images)

--- a/inference.py
+++ b/inference.py
@ -69,8 +69,6 @@ def main(args):
    inference_data = InferenceDataset(args.folder)
    inference_loader = create_inference_loader(inference_data)

-    embed()
-    quit()
    dataset_name = Path(args.folder).name

    infere_model(inference_loader, model, dataset_name)