【python】python+numpy模块读、写raw图并使用opencv显示图片

参考链接：

1. 使用Python读取raw格式图像并显示
2. NumPy 数据类型

---

python对raw图的操作，读取raw图、保存raw图，raw10转raw8操作。

大概用到以下函数：

1. astype()-进行数值类型转换
2. numpy.reshape()-将数组重新排列
3. numpy.fromfile()-读取文件，将文件中的数据以numpy.ndarray类型保存
4. numpy.ndarray.tofile()把numpy.ndarray数据保存到文件中；

---

numpy中的常用数据类型：

---

参考代码：

import numpy as np

def read_raw(file:str,shape:tuple,dtype):
    '''
    读取raw图
    :param file: 文件名
    :param shape: 读取的数据排列，(row,col,channel)
    :param dtype: raw文件类型
    :return: 读取的数据
    '''
    # 从raw文件中读取数据
    data = np.fromfile(file,dtype=dtype)
    # 将读取到的数据重新排列
    data = np.reshape(data,newshape=shape)
    # 返回数据
    return data


def write_raw(file:str,data:np.ndarray):
    '''
    保存raw图
    :param file: 文件名
    :param data: 保存的数据
    :return: 无返回值
    '''
    data.tofile(file)  # 保存数据data到文件file中


def raw_to_raw8(data:np.ndarray,divide:int):
    '''
    raw10、raw12、raw16数据转raw8数据
    :param data: 原始raw数据，pixel raw格式
    :param divide: 转换系数，raw10是4，raw12是power(2,4),raw16是power(2,8)
    :return: 返回raw8数据
    '''
    new_data = data/divide
    return new_data.astype(np.uint8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

实际测试：

if __name__ == "__main__":
    img = read_raw(str("crop"+"."+"raw"),(400,534,1),dtype=np.uint16)
    print(type(img))
    print(np.shape(img))

    write_raw("4.raw",img)
    new_data = raw_to_raw8(img,4)
    write_raw("5.raw",new_data)

    cv2.imshow("raw",new_data)
    dst = cv2.cvtColor(new_data,cv2.COLOR_BayerGR2BGR)
    print(np.shape(dst))
    cv2.imshow("bmp",dst)
    cv2.waitKey(0)
1
2
3
4
5
6
7
8
9
10
11
12
13
14

显示结果，raw是单通道是，所以显示是黑白的，bmp是三通道，显示是彩色的。