`allow_empty=True` for Classify dataset class directories (#14301)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
7 months ago · 7a6bb2b1c9
parent 8ef1d81e66
commit 7a6bb2b1c9
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/data/dataset.py
+++ b/ultralytics/data/dataset.py
@ -416,7 +416,7 @@ class ClassificationDataset:
        import torchvision  # scope for faster 'import ultralytics'

        # Base class assigned as attribute rather than used as base class to allow for scoping slow torchvision import
-        self.base = torchvision.datasets.ImageFolder(root=root)
+        self.base = torchvision.datasets.ImageFolder(root=root, allow_empty=True)
        self.samples = self.base.samples
        self.root = self.base.root