PIL image to array
2017-05-11 本文已影响179人
想学会飞行的阿番
暂时找到两种方法:
>>> import Image
>>> import numpy as np
>>> img = Image.open(img_dir)
#方法一
>>> img_arr_1 = np.array(list(img.getdata()))
>>> img_arr_1 = img_arr_1.reshape((img.size[1],img.size[0],3))
#方法二
>>> img_arr_2 = np.fromstring(img.tobytes(),dtype=np.uint8)
>>> img_arr_2 = img_arr_2.reshape((img.size[1],img.size[0],3))
>>>
这样就可以愉快的放到tf.placeholder里面使用啦
注:方法二比方法一快很多
参考资料