本文记录一下TensorFLow的几种图片读取方法,官方文档有较为全面的介绍。转自脚本之家
1.使用gfile读图片,decode输出是Tensor,eval后是ndarray
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
|
输出为:
1.3.0
< class 'bytes'>
< class 'tensorflow.python.framework.ops.Tensor'>
< class 'numpy.ndarray'>
(666, 1000, 3)
uint8
图片显示(略)
2.使用WholeFileReader输入queue,decode输出是Tensor,eval后是ndarray
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
|
3.使用read_file,decode输出是Tensor,eval后是ndarray
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
|
输出是:
1.3.0
< class 'tensorflow.python.framework.ops.Tensor'>
< class 'tensorflow.python.framework.ops.Tensor'>
< class 'numpy.ndarray'>
(666, 1000, 3)
uint8
显示图片(略)
4.TFRecords:
有空再看。
如果图片是根据分类放在不同的文件夹下,那么可以直接使用如下代码:
//www.jb51.net/article/134532.htm
//www.jb51.net/article/134539.htm
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。