Skip to content

Commit 6c00fa5

Browse files
committed
updated reader_factory to correct extra folders
1 parent e44f14d commit 6c00fa5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/data/readers/reader_factory.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def create_reader(
3939
# default fallback path (backwards compat), use image tar if root is a .tar file, otherwise image folder
4040
# FIXME support split here or in reader?
4141
if os.path.isfile(root) and os.path.splitext(root)[1] == '.tar':
42-
reader = ReaderImageInTar(root, **kwargs)
42+
reader = ReaderImageInTar(root + '/' + split, **kwargs)
4343
else:
44-
reader = ReaderImageFolder(root, **kwargs)
44+
reader = ReaderImageFolder(root + '/' + split, **kwargs)
4545
return reader

0 commit comments

Comments
 (0)