如何将二进制 dat 文件 (.dat) 保存为 png 或 jpg 格式？

Question

我有一个二进制文件格式 .dat。我想将其转换为 png 或 jpg 格式，我尝试使用 numpy.loadtxt() 对其进行解码以获取数组列表，然后将其保存为 png 或 jpg 格式。但是当我执行它时它给出了错误

文件“C:\Users\SETIKO\Desktop\test\test6-savedattopng.py”，第 13 行，在 arr = np.loadtxt(f)
文件“C:\Users\SETIKO\AppData\Local\Programs\Python\Python35\lib\site-packages\numpy\lib\npyio.py”，第 1101 行，在 loadtxt for x in read_data(_loadtxt_chunksize):
文件“C:\Users\SETIKO\AppData\Local\Programs\Python\Python35\lib\site-packages\numpy\lib\npyio.py”，第 1028 行，在 read_data items = [conv(val) for (conv, val) in zip(converters, vals)]
文件“C:\Users\SETIKO\AppData\Local\Programs\Python\Python35\lib\site-packages\numpy\lib\npyio.py”，第 1028 行，在 items = [conv(val) for (conv, val) in zip(converters, vals)]
文件“C:\Users\SETIKO\AppData\Local\Programs\Python\Python35\lib\site-packages\numpy\lib\npyio.py”，第 746 行，在 floatconv 返回浮点数（x）

这是我使用的完整代码

import numpy as np

f = open('gt.dat', 'rb')
arr = np.loadtxt(f)

我的代码中是否缺少某些内容？或者也许有人有其他想法而不必使用numpy.loadtxt()？

这里是我使用的二进制文件： Binary dat file

Answer 1

最简单的方法是，您可以读取文件并将其制成 PIL/枕头图像，如下所示：

import numpy as np
from PIL import Image

# Read file into Numpy array as uint8
na = np.fromfile('GT.dat', dtype=np.uint8)

# Reshape to 5292 px wide, 744 px tall and RGBA, i.e. 4 bytes/per pixel
na = np.reshape(na,(744,5292,4))

# Drop the alpha channel, retaining just RGB
na = na[:,:,:3]

# Convert to PIL/Pillow Image - whilst also brightening because values are all <=5
im = Image.fromarray(na*50,'RGB') 

# Save
im.save('result.png')

---

或者您根本不用编写任何代码，只需使用大多数 Linux 发行版中包含的 ImageMagick，它可用于 macOS 和 Windows。所以，就在终端：

convert -depth 8 -size 5292x744 rgba:GT.dat -normalize result.png

---

更新答案

似乎所有数据实际上都在红色通道中，而在绿色、蓝色或 Alpha 通道中什么都没有。因此，我们将丢弃这些，然后，由于红色通道中的值都是小整数，我们将创建一个调色板图像，其中这些整数是我们可以轻松操作的调色板的索引。

#!/usr/bin/env python3

import numpy as np
from PIL import Image

# Read file into Numpy array as uint8
na = np.fromfile('GT.dat', dtype=np.uint8)

# Reshape to 5292 px wide, 744 px tall and RGBA, i.e. 4 bytes/per pixel
na = np.reshape(na,(744,5292,4))

# Construct a palette
palette = [  
    0, 0, 0,      # 0 - black 
    255, 0, 0,    # 1 - red 
    0, 255, 0,    # 2 - green 
    0, 0, 255     # 3 - blue 
]

# There must be exactly 256 RGB colours, so pad palette to length 768 with zeroes
palette += (768-len(palette))*[0]

# Drop everything except the Red channel which contains all the data
na = na[:,:,0]

# Convert to single channel PIL/Pillow Image
im = Image.fromarray(na,'L') 

# Put our palette into image
im.putpalette(palette)

# Save
im.save('result.png')