From 873758d461983af82fcd46d84ca241e1576d00de Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 11:22:59 +0200
Subject: [PATCH] embed for bf

---
 datasets.py  | 3 ++-
 inference.py | 2 --
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/datasets.py b/datasets.py
index 85c5433..43354c5 100644
--- 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)
 
diff --git a/inference.py b/inference.py
index 302d6f8..22ed687 100644
--- 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)