在tensorflow session.run()中输入几个图像

时间:2019-06-03 11:41:21

标签: python tensorflow

我有一个代码使用tensorflow通过GPU分析一些图像,我使用tf.session并将模型的session.run中的帧引入为

with tf.Session() as sess:
    res, img = cap.read()
    heatmaps_result, offsets_result, displacement_fwd_result, displacement_bwd_result = sess.run(model_outputs,feed_dict={'image:0': my_batch[0]})

当我在GPU上运行它时,它具有可用空间,因此我想介绍几个图像。我尝试了Bach several images in tensorflow中所说的内容,但收到错误消息:

“名称'image_tensor:0'表示不存在的张量。图中的操作'image_tensor'不存在。”

我在巴赫中将图像介绍为:

cnt = 0
my_batch = list()
while (cnt < 5):
    cnt = cnt + 1
    res, img = cap.read()
    if not res:
        break
    my_batch.append(img)

1 个答案:

答案 0 :(得分:0)

是否存在一个名为'image:0'的占位符,因为在使用feed_dict时,您需要输入与输入占位符相对应的名称。