change to friendly error message

apache · Jul 1, 2018 · 50e6b02 · 50e6b02
1 parent 37c8165
commit 50e6b02
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/mxnet/image/image.py b/python/mxnet/image/image.py
@@ -133,6 +133,8 @@ def imdecode(buf, *args, **kwargs):
     <NDArray 224x224x3 @cpu(0)>
     """
     if not isinstance(buf, nd.NDArray):
+        if sys.version_info[0] == 3 and not (isinstance(buf, bytes) or isinstance(buf, np.ndarray)):
+            raise ValueError('buf must bytes or numpy.ndarray,if you input str,please convert it to bytes')
         buf = nd.array(np.frombuffer(buf, dtype=np.uint8), dtype=np.uint8)
     return _internal._cvimdecode(buf, *args, **kwargs)