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里面使用啦
注:方法二比方法一快很多
参考资料

上一篇下一篇

猜你喜欢

热点阅读