Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
1dc5724a
"vscode:/vscode.git/clone" did not exist on "0d794af6a5730efcfb23aa898232c6df3afec347"
Commit
1dc5724a
authored
Jan 21, 2017
by
Alykhan Tejani
Browse files
add MNIST dataset to docs. Improve formatting of LSUN docs
parent
6f342d3a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
22 additions
and
6 deletions
+22
-6
README.rst
README.rst
+22
-6
No files found.
README.rst
View file @
1dc5724a
...
...
@@ -40,6 +40,7 @@ Datasets
The following dataset loaders are available:
- `MNIST <#mnist>`__
- `COCO (Captioning and Detection) <#coco>`__
- `LSUN Classification <#lsun>`__
- `ImageFolder <#imagefolder>`__
...
...
@@ -66,6 +67,21 @@ but they all take the keyword args:
transforms it. For example, take in the caption string and return a
tensor of word indices.
MNIST
~~~~~
``dset.MNIST(root, train=True, transform=None, target_transform=None, download=False)``
``root``: root directory of dataset where ``processed/training.pt`` and ``training/test.pt`` exist
``train``: ``True`` - use training set, ``False`` - use test set.
``transform``: transform to apply to input images
``target_transform``: transform to apply to targets (class labels)
``download``: whether to download the MNIST data
COCO
~~~~
...
...
@@ -115,12 +131,12 @@ LSUN
``dset.LSUN(db_path, classes='train', [transform, target_transform])``
- db
\
_path = root directory for the database files
- classes =
- 'train' - all categories, training set
- 'val' - all categories, validation set
- 'test' - all categories, test set
- ['bedroom
\
_train'
,
'church
\
_train', ...] : a list of categories to
-
``
db_path
``
= root directory for the database files
-
``
classes
``
=
-
``
'train'
``
- all categories, training set
-
``
'val'
``
- all categories, validation set
-
``
'test'
``
- all categories, test set
- [
``
'bedroom_train'
``, ``
'church_train'
``
, ...] : a list of categories to
load
CIFAR
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment