HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Alexis IMBERT 2024-07-10 01:02:10 +02:00 • committed by GitHub
parent 8ef1d81e66
commit 7a6bb2b1c9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/data/dataset.py
View file

@ -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