mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix compatible BUG: QMNIST and ImageNet datasets do not exist in torchvision 0.10.1.

Browse Source
This commit is contained in:
hongxin xiang 2022-11-24 14:37:44 +08:00
parent 25ffac6880
commit bdc9fad638
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
timm/data/dataset_factory.py
View File

@ -4,7 +4,7 @@ Hacked together by / Copyright 2021, Ross Wightman
""" """
import os import os
from torchvision.datasets import CIFAR100, CIFAR10, MNIST, QMNIST, KMNIST, FashionMNIST, ImageNet, ImageFolder from torchvision.datasets import CIFAR100, CIFAR10, MNIST, KMNIST, FashionMNIST, ImageFolder
try: try:
from torchvision.datasets import Places365 from torchvision.datasets import Places365
has_places365 = True has_places365 = True
@ -15,6 +15,16 @@ try:
has_inaturalist = True has_inaturalist = True
except ImportError: except ImportError:
has_inaturalist = False has_inaturalist = False
try:
from torchvision.datasets import QMNIST
has_qmnist = True
except ImportError:
has_qmnist = False
try:
from torchvision.datasets import ImageNet
has_imagenet = True
except ImportError:
has_imagenet = False
from .dataset import IterableImageDataset, ImageDataset from .dataset import IterableImageDataset, ImageDataset
@ -22,7 +32,6 @@ _TORCH_BASIC_DS = dict(
cifar10=CIFAR10, cifar10=CIFAR10,
cifar100=CIFAR100, cifar100=CIFAR100,
mnist=MNIST, mnist=MNIST,
qmist=QMNIST,
kmnist=KMNIST, kmnist=KMNIST,
fashion_mnist=FashionMNIST, fashion_mnist=FashionMNIST,
) )