python/numpy读取二进制文件

需求

我有一个文件，内容是摄像头的BGR数据，分辨率是1920X1080， 每个像素3个字节，共6220800字节。我已经写到磁盘上的文件里，需要用python读取到numpy中进行处理。

方法1：np.frombuffer

使用open打开文件，二进制读取后用np.frombuffer转换为numpy.ndarray。

import cv2
import os
import numpy as np

filepath='1'

f = open(filepath, 'rb')
data_rd = f.read()
print(f"type(data_rd)={type(data_rd)}, len(data_rd)={len(data_rd)}")
mat_rd = np.frombuffer(data_rd, dtype=np.uint8)
print(f"type(mat_rd)={type(mat_rd)}, len(mat_rd)={len(mat_rd)}")

mat = mat_rd.reshape((1080, 1920, 3))  #
print(f"mat.shape={mat.shape}, mat.size={mat.size}")

cv2.imshow("Demo1", mat)
key = cv2.waitKey(0)  # 等待按键命令, 1000ms 后自动关闭

命令行输出为：

type(data_rd)=<class 'bytes'>, len(data_rd)=6220800
type(mat_rd)=<class 'numpy.ndarray'>, len(mat_rd)=6220800
mat.shape=(1080, 1920, 3), mat.size=6220800

方法2：np.fromfile

使用numpy的fromfile函数直接从磁盘读取。

import cv2
import os
import numpy as np

filepath='1'

mat = np.fromfile('1', dtype=np.uint8)
print(f"mat.shape={mat.shape}, mat.size={mat.size}")
print(f"type(mat)={type(mat)}")

mat = mat.reshape((1080, 1920, 3))  #
print(f"mat.shape={mat.shape}, mat.size={mat.size}")

cv2.imshow("Demo1", mat)
key = cv2.waitKey(0)  # 等待按键命令

命令行输出：

mat.shape=(6220800,), mat.size=6220800
type(mat)=<class 'numpy.ndarray'>
mat.shape=(1080, 1920, 3), mat.size=6220800

参考资料

Python读取二进制文件：深入解析与技术实现

python中如何将bytes格式转换为numpy ndarray