使用numpy.fromfile读取raw图像文件

最新推荐文章于 2024-10-10 14:52:51 发布

考拉不是大叔

最新推荐文章于 2024-10-10 14:52:51 发布

阅读量5.3k

点赞数 2

分类专栏： numpy整理文章标签： numpy python

本文链接：https://blog.csdn.net/kevin198528/article/details/107708264

版权

numpy整理专栏收录该内容

2 篇文章 0 订阅

订阅专栏

以前用的是这套接口，总感觉慢腾腾的，像老大爷散步

def bin2numpy(file_path, shape):
    pic_gray = np.zeros(shape)
    with open(file_path, "rb") as f:
        for i in range(shape[0]):
            for j in range(shape[1]):
                data = f.read(2)
                pix = struct.unpack('H', data)[0]
                pic_gray[i, j] = pix

    return pic_gray

写个测试程序猛然发现读一张1920x1080的raw格式图片要nm
一秒钟。。。
一秒。。
一。

1.019157886505127

果断换用np.fromfile接口

def bin2numpy_ex(file_path, shape):
    rawImg = np.fromfile(file_path, dtype=np.uint16)
    rawImg = rawImg[: shape[0]*shape[1]]
    pic_gray = rawImg.reshape(shape)

    return pic_gray