mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge 6c00fa52b4 into c8c4f256b8

pull/2448/merge
Rorry Brenner 2025-04-22 19:49:59 -04:00 committed by GitHub
parent c8c4f256b8 6c00fa52b4
commit 06309819fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/data/readers/reader_factory.py
View File

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