tf.image.encode_png给出全黑输出

Question

所以基本上我正在使用tfr1.9。我正在使用tf.dataset api导入我的数据。 我正在尝试使用以下代码将png文件编码为输出，但是我所有的输出图像都变成黑色。

    output = Vgg16.output
    logits = tf.cast(tf.to_int32(output>0.5),tf.uint8)
    predictions= tf.multiply(logits, 255)
    write_op  = tf.map_fn(lambda image:tf.image.encode_png(image),predictions,parallel_iterations=6,dtype=tf.string

然后在tf中添加以下代码以编写

 if(epoch%100==0):
            sess.run(images_iterator_init_op)
            segmented_output = sess.run(write_op)
            i=0
            if not (os.path.exists(result_path+'train/'+str(epoch)+'/')):
                os.makedirs(result_path+'train/'+str(epoch) +'/')
            for image in segmented_output:
                with open(result_path+'train/'+str(epoch)+'/'+str(i)+'.png','w') as f:
                    f.write(image)
                    i = i+1