This is an automated email from the ASF dual-hosted git repository.
kellen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new afa4c3a Correct shapes of images in cifar10 and cifar100 (#13348)
afa4c3a is described below
commit afa4c3a85f0684e0dbdb097c9ffb4b5d14f9ef9d
Author: Masaki Samejima <[email protected]>
AuthorDate: Mon Nov 26 13:14:29 2018 +0900
Correct shapes of images in cifar10 and cifar100 (#13348)
* Correct shapes of images in cifar10 and cifar100
cifar10 and cifar100 have 3 channels
* Retrigger build
---
python/mxnet/gluon/data/vision/datasets.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python/mxnet/gluon/data/vision/datasets.py
b/python/mxnet/gluon/data/vision/datasets.py
index 267aa1e..12ef7e1 100644
--- a/python/mxnet/gluon/data/vision/datasets.py
+++ b/python/mxnet/gluon/data/vision/datasets.py
@@ -128,7 +128,7 @@ class FashionMNIST(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 @@ class CIFAR10(dataset._DownloadedDataset):
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
----------
