Skip to content

Commit be3ef03

Browse files
datumboxfacebook-github-bot
authored andcommitted
[fbsync] add prototype imagenet dataset (#4640)
Summary: * 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 Reviewed By: NicolasHug Differential Revision: D31916331 fbshipit-source-id: 38a598f951923342e488f0188f40c74d5b13108c
1 parent 7afdb1f commit be3ef03

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)