KellenSunderland closed pull request #13348: Correct shapes of images in cifar10 and cifar100 URL: https://github.com/apache/incubator-mxnet/pull/13348
This is a PR merged from a forked repository. As GitHub hides the original diff on merge, it is displayed below for the sake of provenance: As this is a foreign pull request (from a fork), the diff is supplied below (as it won't show otherwise due to GitHub magic): diff --git a/python/mxnet/gluon/data/vision/datasets.py b/python/mxnet/gluon/data/vision/datasets.py index 0e380c06efa..ff9a84f1b3e 100644 --- a/python/mxnet/gluon/data/vision/datasets.py +++ b/python/mxnet/gluon/data/vision/datasets.py @@ -128,7 +128,7 @@ def __init__(self, root=os.path.join(base.data_dir(), 'datasets', 'fashion-mnist class CIFAR10(dataset._DownloadedDataset): """CIFAR10 image classification dataset from https://www.cs.toronto.edu/~kriz/cifar.html - Each sample is an image (in 3D NDArray) with shape (32, 32, 1). + Each sample is an image (in 3D NDArray) with shape (32, 32, 3). Parameters ---------- @@ -190,7 +190,7 @@ def _get_data(self): class CIFAR100(CIFAR10): """CIFAR100 image classification dataset from https://www.cs.toronto.edu/~kriz/cifar.html - Each sample is an image (in 3D NDArray) with shape (32, 32, 1). + Each sample is an image (in 3D NDArray) with shape (32, 32, 3). Parameters ---------- ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services