如何在Python中將“.png”深度圖像轉換為“.pfm”？

Question

我有來自 PNG 格式數據集的深度圖像，但我需要使用的神經網絡接受 PFM 格式深度圖像。 我在 inte.net 上搜索過，但找不到任何正確的答案和解釋。

PNG 深度圖像有 3 個通道，形狀為 640x480x3 16 位深度圖像。 我需要將此 PNG 深度圖像轉換為 960x540 的 PFM 圖像。

要轉換為 PFM 的 PNG 深度圖像

圖像值和形狀，打印為 Numpy 數組

我使用 OpenCV 讀取 PNG 文件，然后嘗試將它們保存為 PFM 格式，該格式給出了 Null 值。

Answer 1

首先，從 OpenCV 讀取深度圖像為 Unchanged。

depth_PNG = cv2.imread('imageName.png', cv2.IMREAD_UNCHANGED)

只需使用 cv2.imwrite 以所需格式保存深度圖像。

depth_PFM = cv2.imwrite('newName.pfm', depth_PNG)