Skip to content

Commit 58f313b

Browse files
authored
add prototype imagenet dataset (#4640)
* add prototype imagenet dataset * add missing checksums * fix mypy * add human readable categories * cleanup * sort categories ascending based on wnid * remove accidentally added file * cleanup category file generation * fix mypy
1 parent 588c698 commit 58f313b

File tree

6 files changed

+1290
-95
lines changed

6 files changed

+1290
-95
lines changed

torchvision/prototype/datasets/_builtin/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from .celeba import CelebA
33
from .cifar import Cifar10, Cifar100
44
from .coco import Coco
5+
from .imagenet import ImageNet
56
from .mnist import MNIST, FashionMNIST, KMNIST, EMNIST, QMNIST
67
from .sbd import SBD
78
from .voc import VOC

0 commit comments

Comments
 (0)