From f3e88078c4dfabcd332c0ae12b5298a0b7a7ba71 Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 27 Oct 2023 11:25:32 +0200
Subject: [PATCH] bf in InferenceDataset in datasets.py

---
 datasets.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/datasets.py b/datasets.py
index 43354c5..138841f 100644
--- a/datasets.py
+++ b/datasets.py
@@ -24,15 +24,11 @@ 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)
 
     def __getitem__(self, idx):
-        image_name = self.all_images[idx]
-        image_path = os.path.join(self.dir_path, image_name)
-
+        image_path = self.all_images[idx]
         img = Image.open(image_path)
         img_tensor = F.to_tensor(img.convert('RGB'))