You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by jx...@apache.org on 2018/01/19 00:35:16 UTC
[incubator-mxnet] 09/19: fix (#8736)
This is an automated email from the ASF dual-hosted git repository.
jxie pushed a commit to branch vision
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 7996da927ee269fb157531685e5affa2fd608cc7
Author: Eric Junyuan Xie <pi...@users.noreply.github.com>
AuthorDate: Mon Nov 20 22:06:32 2017 -0800
fix (#8736)
---
python/mxnet/gluon/data/dataset.py | 2 +-
python/mxnet/gluon/data/vision/datasets.py | 6 +++---
python/mxnet/gluon/data/vision/transforms.py | 2 +-
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/python/mxnet/gluon/data/dataset.py b/python/mxnet/gluon/data/dataset.py
index 35d4c5c..f7ab395 100644
--- a/python/mxnet/gluon/data/dataset.py
+++ b/python/mxnet/gluon/data/dataset.py
@@ -18,7 +18,7 @@
# coding: utf-8
# pylint: disable=
"""Dataset container."""
-__all__ = ['Dataset', 'SimpleDataset', 'ArrayDataset', 'LabeledDataset',
+__all__ = ['Dataset', 'SimpleDataset', 'ArrayDataset',
'RecordFileDataset']
import os
diff --git a/python/mxnet/gluon/data/vision/datasets.py b/python/mxnet/gluon/data/vision/datasets.py
index cb40af5..4ddc2e3 100644
--- a/python/mxnet/gluon/data/vision/datasets.py
+++ b/python/mxnet/gluon/data/vision/datasets.py
@@ -28,9 +28,9 @@ import struct
import warnings
import numpy as np
-from . import dataset
-from ..utils import download, check_sha1
-from ... import nd, image, recordio
+from .. import dataset
+from ...utils import download, check_sha1
+from .... import nd, image, recordio
apache_repo_url = 'https://apache-mxnet.s3-accelerate.dualstack.amazonaws.com/'
diff --git a/python/mxnet/gluon/data/vision/transforms.py b/python/mxnet/gluon/data/vision/transforms.py
index fa7c0f2..e1deef6 100644
--- a/python/mxnet/gluon/data/vision/transforms.py
+++ b/python/mxnet/gluon/data/vision/transforms.py
@@ -58,7 +58,7 @@ class ToTensor(HybridBlock):
super(ToTensor, self).__init__()
def hybrid_forward(self, F, x):
- return F.cast(x, 'float32').transpose((2, 0, 1))
+ return F.image.to_tensor(x)
class Normalize(HybridBlock):
--
To stop receiving notification emails like this one, please contact
"commits@mxnet.apache.org" <co...@mxnet.apache.org>.