From 6c00fa52b4b645a8275a5cd2e672254cf2140ac3 Mon Sep 17 00:00:00 2001 From: Rorry Date: Sun, 2 Mar 2025 22:09:03 -0500 Subject: [PATCH] updated reader_factory to correct extra folders --- timm/data/readers/reader_factory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/data/readers/reader_factory.py b/timm/data/readers/reader_factory.py index cfe1910f5a..9da49707b6 100644 --- a/timm/data/readers/reader_factory.py +++ b/timm/data/readers/reader_factory.py @@ -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