在Tensorflow中将图像写入文件?

时间:2017-10-23 07:51:12

标签: python tensorflow

我正在尝试在此函数中插入一行,这样当它解码jpeg时,它会将jpeg写入文件(我这样做是出于完整性检查的目的)。

def decoded_jpeg(image_buffer, scope=None):
    with tf.name_scope(values=[image_buffer], name=scope,
                 default_name='decode_jpeg'):

        fname = tf.constant('2.jpg')
        image_buffer = tf.write_file(fname, image_buffer)
        image = tf.image.decode_jpeg(image_buffer, channels=3)

        image = tf.image.convert_image_dtype(image, dtype=tf.float32)
        return image

上面的代码没有错误输出,但它没有所需的效果,因为它tf.write_file返回一个op而不是一个张量字符串。我想知道如何重写上面的代码以使其正常工作。

其他代码可以在这里看到(它是Inception CNN):https://github.com/tensorflow/models/tree/master/research/inception/inception

0 个答案:

没有答案