set path to masks from one folder

hasibzunair · Oct 13, 2022 · 59d6004 · 59d6004
1 parent b820fc3
commit 59d6004
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dataset.py b/dataset.py
@@ -35,7 +35,7 @@ def build_dataset(self):
             self.test_images = sorted(glob.glob(self._input_folder + "/*.png"))
             self.test_labels = sorted(glob.glob(self._label_folder + "/*.png"))
 
-        self._scribbles_folder = os.path.join(self._data_folder, 'SCRIBBLES')
+        self._scribbles_folder = './datasets/SCRIBBLES'
         self._scribbles = sorted(glob.glob(self._scribbles_folder + "/*.png"))[::-1][:1000] # For heavy masking [::-1]
 
     def __len__(self):
@@ -105,7 +105,7 @@ def build_dataset(self):
             self.test_images = sorted(glob.glob(self._input_folder + "/*.png"))
             self.test_labels = sorted(glob.glob(self._label_folder + "/*.png"))
 
-        self._scribbles_folder = os.path.join(self._data_folder, 'SCRIBBLES')
+        self._scribbles_folder = './datasets/SCRIBBLES'
         self._scribbles = sorted(glob.glob(self._scribbles_folder + "/*.png"))[::-1][:1000] # For heavy masking [::-1]
 
     def __len__(self):
@@ -180,7 +180,7 @@ def build_dataset(self):
             self.test_images = sorted(glob.glob(self._input_folder + "/*.png"))
             self.test_labels = sorted(glob.glob(self._label_folder + "/*.png"))
 
-        self._scribbles_folder = os.path.join(self._data_folder, "TrainDataset", 'SCRIBBLES')
+        self._scribbles_folder = './datasets/SCRIBBLES'
         self._scribbles = sorted(glob.glob(self._scribbles_folder + "/*.png"))[::-1][:1000] # For heavy masking [::-1]
 
     def __len__(self):