美文网首页
PIL image to array

PIL image to array

作者: 想学会飞行的阿番 | 来源:发表于2017-05-11 17:46 被阅读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里面使用啦
注:方法二比方法一快很多
参考资料

相关文章

网友评论

      本文标题:PIL image to array

      本文链接:https://www.haomeiwen.com/subject/jqootxtx.html