[MXNET-360]auto convert str to bytes in img.imdecode when py3 (#10697)

yajiedesign · zhreshold · commit d4991a0e324b · 2018-09-21T10:49:56.000-07:00
* change to friendly error message

* fix lint

* fix lint

* fix lint

* update raise message
diff --git a/python/mxnet/image/image.py b/python/mxnet/image/image.py
@@ -21,13 +21,15 @@
 
 from __future__ import absolute_import, print_function
 
+import sys
 import os
 import random
 import logging
 import json
 import warnings
 import numpy as np
 
+
 try:
     import cv2
 except ImportError:
@@ -133,6 +135,9 @@ 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, np.ndarray)):
+            raise ValueError('buf must be of type bytes or numpy.ndarray,'
+                             'if you would like to input type str, please convert to bytes')
         buf = nd.array(np.frombuffer(buf, dtype=np.uint8), dtype=np.uint8)
     return _internal._cvimdecode(buf, *args, **kwargs)
 
