From 7a6bb2b1c9dc7178e266cc6b2c47522d35b8196c Mon Sep 17 00:00:00 2001 From: Alexis IMBERT <97242148+Alexis-IMBERT@users.noreply.github.com> Date: Wed, 10 Jul 2024 01:02:10 +0200 Subject: [PATCH] `allow_empty=True` for Classify dataset class directories (#14301) Co-authored-by: Glenn Jocher --- ultralytics/data/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/data/dataset.py b/ultralytics/data/dataset.py index 8bbb08f8a..cd50c21c7 100644 --- 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