Question

根据doc图像被规范化以适合[0,255]范围。我的模型有时会生成深色图像（范围x <255，例如[0，200]）。然后将此类图像转换为[0,255]，因此它们非常明亮。停止此操作的唯一方法是在图像一角添加255像素。有没有办法避免张量板上的这种缩放？我正在自己缩放所有内容。

Answer 1

如果您已经拥有float[16,32,64]个dtypes中的一个图像批处理，那么您对缩放的比例感到满意，只需在进行摘要之前将其强制转换为uint8即可，缩放比例将被保留。 / p>

byte_image = tf.cast(image, tf.uint8)
tf.summary.image(name=name, tensor=byte_image)

您可能会发现将浮点图像保留在[0, 1)中并使用tf.image.convert_image_dtype函数在不同的dtypes而非tf.cast之间进行转换对于保持一致性很有帮助。