如何在 OpenCV2 中将 float32 图片写入视频文件？

Question

我在这里遇到了一个奇怪的问题。 我从我的神经网络中取回了一个输出图像，它具有 dtype float32。我添加了一张从互联网上获取的背景图片，将其从 uint8 转换为 float32，然后将我的输出添加到背景图片之上。 现在，我想保存它，但 cv2.VideoWriter.write(blahblah) 只接受 uint8。我认为要将 float32 转换为 uint8，只需执行 * 255，但这可能是错误的，因为当我这样做时会遇到一些奇怪的眩光（我的照片可以在这里找到：OpenCV image shows weird glare after un-normalizing with * 255） ，可能是因为浮点经典错误或这种性质的事情。那么，如何安全地将 float32 转换为 uint8？或者，如何将 float32 写入文件？

说实话，我对一些事情感到很困惑。我知道 uint8 是 [0, 255] 而 float32 是 [0, 1]，我知道 cv2.videowriter.write 接受 uint8，就是这样。附带说明一下，一个解决方案似乎是从一开始就将所有内容更改为 .astype(uint8) ，但我的神经网络吐出了一个 float32 ，所以我必须使用它。

Answer 1

Float32 没有设置从 0 到 1，这很可能由您的神经网络完成（softmax 或 sigmoid 作为最后一个激活函数）。要将某些内容设置为不同的范围，可以使用：

normalized_array = (array - np.min(array))/(np.max(array) - np.min(array)) # this set the range from 0 till 1
img_array = (normalized_array * 255).astype(np.uint8) # set is to a range from 0 till 255