diff --git a/docs/source/datasets.rst b/docs/source/datasets.rst index 7fc7c5da1ca..a4a7fe57ad6 100644 --- a/docs/source/datasets.rst +++ b/docs/source/datasets.rst @@ -30,6 +30,11 @@ MNIST .. autoclass:: MNIST +Fashion-MNIST +~~~~~~~~~~~~~ + +.. autoclass:: FashionMNIST + COCO ~~~~ diff --git a/torchvision/datasets/mnist.py b/torchvision/datasets/mnist.py index dee28287bba..44ade5d4378 100644 --- a/torchvision/datasets/mnist.py +++ b/torchvision/datasets/mnist.py @@ -140,7 +140,20 @@ def download(self): class FashionMNIST(MNIST): - """`Fashion MNIST `_ Dataset. + """`Fashion-MNIST `_ Dataset. + + Args: + root (string): Root directory of dataset where ``processed/training.pt`` + and ``processed/test.pt`` exist. + train (bool, optional): If True, creates dataset from ``training.pt``, + otherwise from ``test.pt``. + download (bool, optional): If true, downloads the dataset from the internet and + puts it in root directory. If dataset is already downloaded, it is not + downloaded again. + transform (callable, optional): A function/transform that takes in an PIL image + and returns a transformed version. E.g, ``transforms.RandomCrop`` + target_transform (callable, optional): A function/transform that takes in the + target and transforms it. """ urls = [ 'http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gz',