在python中显示二进制文件中的数据

时间:2013-11-27 15:21:52

标签: python python-3.x numpy matplotlib

我有2000个图像存储为单个二进制文件“file.dat”和一个512字节的头部到此文件。每个图像的格式为512 * 512 * 2字节(unsigned int 16)。我的任务是将所有这些图像可视化为视频。我怎么能在python中这样做?我的问题是从阅读图像序列开始。我是python中的新手。

1 个答案:

答案 0 :(得分:1)

Numpy对于以简单的二进制文件格式阅读非常方便。

从它的声音中,你有一个uin16的大型二进制文件,你想要读入3D数组并进行可视化。我们不必将它全部加载到内存中,但是对于这个例子,我们将会。

以下是代码外观的基本概念:

import numpy as np
import matplotlib.pyplot as plt

def main():
    data = read_data('test.dat', 512, 512)
    visualize(data)

def read_data(filename, width, height):
    with open(filename, 'r') as infile:
        # Skip the header
        infile.seek(512)
        data = np.fromfile(infile, dtype=np.uint16)
    # Reshape the data into a 3D array. (-1 is a placeholder for however many
    # images are in the file... E.g. 2000)
    return data.reshape((width, height, -1))

def visualize(data):
    # There are better ways to do this, but let's keep it simple
    plt.ion()
    fig, ax = plt.subplots()
    im = ax.imshow(data[:,:,0], cmap=plt.cm.gray)
    for i in xrange(data.shape[-1]):
        image = data[:,:,i]
        im.set(data=image, clim=[image.min(), image.max()])
        fig.canvas.draw()

main()