@@ -40,6 +40,7 @@ Datasets
40
40
41
41
The following dataset loaders are available:
42
42
43
+ - `MNIST <#mnist >`__
43
44
- `COCO (Captioning and Detection) <#coco >`__
44
45
- `LSUN Classification <#lsun >`__
45
46
- `ImageFolder <#imagefolder >`__
@@ -66,6 +67,21 @@ but they all take the keyword args:
66
67
transforms it. For example, take in the caption string and return a
67
68
tensor of word indices.
68
69
70
+ MNIST
71
+ ~~~~~
72
+ ``dset.MNIST(root, train=True, transform=None, target_transform=None, download=False) ``
73
+
74
+ ``root ``: root directory of dataset where ``processed/training.pt `` and ``training/test.pt `` exist
75
+
76
+ ``train ``: ``True `` - use training set, ``False `` - use test set.
77
+
78
+ ``transform ``: transform to apply to input images
79
+
80
+ ``target_transform ``: transform to apply to targets (class labels)
81
+
82
+ ``download ``: whether to download the MNIST data
83
+
84
+
69
85
COCO
70
86
~~~~
71
87
@@ -115,12 +131,12 @@ LSUN
115
131
116
132
``dset.LSUN(db_path, classes='train', [transform, target_transform]) ``
117
133
118
- - db \_ path = root directory for the database files
119
- - classes =
120
- - 'train' - all categories, training set
121
- - 'val' - all categories, validation set
122
- - 'test' - all categories, test set
123
- - ['bedroom \_ train', 'church \_ train' , ...] : a list of categories to
134
+ - `` db_path `` = root directory for the database files
135
+ - `` classes `` =
136
+ - `` 'train' `` - all categories, training set
137
+ - `` 'val' `` - all categories, validation set
138
+ - `` 'test' `` - all categories, test set
139
+ - [`` 'bedroom_train' ``, `` 'church_train' `` , ...] : a list of categories to
124
140
load
125
141
126
142
CIFAR
0 commit comments