我正在尝试在此函数中插入一行,这样当它解码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