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

updated reader_factory to correct extra folders

pull/2448/head
Rorry 2025-03-02 22:09:03 -05:00
parent e44f14d7d2
commit 6c00fa52b4
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