From e96e24cfda2381c5f8bc0d3f89c26b8f40129aa1 Mon Sep 17 00:00:00 2001 From: Kethupio Date: Sat, 1 Apr 2023 23:18:29 +0530 Subject: [PATCH] Update imagenet.py --- ldm/data/imagenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/data/imagenet.py b/ldm/data/imagenet.py index 1c473f9c6..64d6db290 100644 --- a/ldm/data/imagenet.py +++ b/ldm/data/imagenet.py @@ -97,7 +97,7 @@ def _load(self): self.relpaths = self._filter_relpaths(self.relpaths) print("Removed {} files from filelist during filtering.".format(l1 - len(self.relpaths))) - self.synsets = [p.split("/")[0] for p in self.relpaths] + self.synsets = [p.split(os.sep)[0] for p in self.relpaths] self.abspaths = [os.path.join(self.datadir, p) for p in self.relpaths] unique_synsets = np.unique(self.synsets)