张量流读取队列中的HDR图像

时间:2018-02-07 07:29:18

标签: tensorflow queue

我正在尝试将HDR图像导入tensorflow进行某些实验。我尝试从队列中读取HDR图像数据库,但似乎tf.reader无法读取此类型的图像。这就是我到目前为止所做的:

filelist_queue = tf.train.string_input_producer(filelist,shuffle=True)

然后我定义一个函数尝试从队列中获取图像:

def gethdrimg(queue): 
    filepath = queue.dequeue
    return imageio.imread(filpath)

我认为这可行,但事实证明返回的文件路径是张量,不能用于imageio.imread。

无论如何,我可以将文件路径作为字符串而不是张量?

1 个答案:

答案 0 :(得分:0)

您可以使用tf.py_func对张量的字符串值运行python函数。

或者您可以使用热切的执行而不用图表(尽管如此,string_input_producer不适用于急切执行)。