代码之家 › 专栏 › 技术社区 › bluesummers

TensorFlow读取和解码一批图像

tensorflow python

bluesummers · 技术社区 · 6 年前

使用 tf.train.string_input_producer 和 tf.image.decode_jpeg 我设法从磁盘读取并解码 单一的 图像。

这是代码:

# -------- Graph
filename_queue = tf.train.string_input_producer(
    [img_path, img_path])

image_reader = tf.WholeFileReader()

key, image_file = image_reader.read(filename_queue)

image = tf.image.decode_jpeg(image_file, channels=3)

# Run my network
logits = network.get_logits(image)

# -------- Session
sess = tf.Session()
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(sess=sess, coord=coord)

logits_output = sess.run(logits)

问题是,当我看到 logit_outputs 即使队列有2个图像,我也只得到1个值。

如何读取和解码整个队列?

1 回复 | 直到 6 年前

benjaminplanche 6 年前

tf.WholeFileReader() ,沿 tf.train.string_input_producer() 作为一个迭代器工作,因此没有一个简单的方法来评估它正在处理的完整数据集的大小。

获取 N 样本出来,你可以用 image_reader.read_up_to(filename_queue, N) .

tf.data 管道:

def _parse_function(filename):
  image_string = tf.read_file(filename)
  image_decoded = tf.image.decode_image(image_string)
  return image_decoded 

# A vector of filenames.
filenames = tf.constant([img_path, img_path])

dataset = tf.data.Dataset.from_tensor_slices((filenames))
dataset = dataset.map(_parse_function).batch(N)
iterator = dataset.make_one_shot_iterator()
next_image_batch = iterator.get_next()

logits = network.get_logits(next_image_batch)
# ...

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

Anshuman Sinha · 将tensforflow中的所有nan值转换为零

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

rob me of my freedom · 在不使用Y_pred和Y_actual的tensorflow中编写自定义损失

2 年前

Chris · 如何捕获导入脚本功能的打印

3 年前

Jay Bhie Santos · 剩余GRU:ValueError:输入具有不兼容的形状。收到的形状(5)和(24)

3 年前

Danijel · C中TensorFlow模型文件的路径?

3 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

3 年前

Arya Anish · tensorflow安装中面临的问题

3 年前

so beautiful memory · 如何将我在本地训练的tensorflow模型正确部署到谷歌云人工智能平台?我部署了它,但没有图像返回

3 年前