Merge branch 'main' into models/convnext_variants

pytorch · Feb 1, 2022 · d708d06 · d708d06
2 parents fd5c99d + 8aad0e0
commit d708d06
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchvision/prototype/datasets/_builtin/dtd.py b/torchvision/prototype/datasets/_builtin/dtd.py
@@ -72,7 +72,8 @@ def _classify_archive(self, data: Tuple[str, Any]) -> Optional[int]:
 
     def _image_key_fn(self, data: Tuple[str, Any]) -> str:
         path = pathlib.Path(data[0])
-        return str(path.relative_to(path.parents[1]))
+        # The split files contain hardcoded posix paths for the images, e.g. banded/banded_0001.jpg
+        return str(path.relative_to(path.parents[1]).as_posix())
 
     def _collate_and_decode_sample(
         self,