Merge branch 'RGB' of github.com:etienneguevel/dinov2 into RGB

2024-05-24 17:13:05 +02:00 · 2024-05-24 17:13:05 +02:00 · 84dd00b098
parent 9ef866789f 2e05e6b4ef
commit 84dd00b098
1 changed files with 3 additions and 0 deletions
--- a/dinov2/data/datasets/custom_image_dataset.py
+++ b/dinov2/data/datasets/custom_image_dataset.py
@ -14,6 +14,9 @@ class ImageDataset(Dataset):
        self.frac = frac
        self.preserved_images = []
        self.images_list = self._get_image_list()
+        self.path_preserved = path_preserved if isinstance(path_preserved, list) else list(path_preserved)
+        self.frac = frac
+        self.preserved_images = []

    def _get_image_list(self):
        images = []