Python 使用 opencv 库将BMP格式图片转RAW

一、安装库的问题

代码中导入包名如下

import cv2

但是 Python Interpreter 应该搜索 opencv-python ,直接搜索 cv2 安装不了

二、代码示例

import cv2

def print_hi(name):
    img = cv2.imread('test/test.bmp', cv2.IMREAD_UNCHANGED)
    img.tofile('test/test.raw')

if __name__ == '__main__':
    print_hi('PyCharm')

　　代码两句搞定，使用 cv2.imread(filepath,flags) 函数读取然后转换即可。不得不说确实方便。

imread 参数含义：

• filepath：读入图片路径
• flags：读入图片的标志 
  • cv2.IMREAD_COLOR：默认参数，读入一副彩色图片，忽略alpha通道
  • cv2.IMREAD_GRAYSCALE：读入灰度图片
  • cv2.IMREAD_UNCHANGED：读入完整图片，包括alpha通道

三、特殊场景

    需要转换后的raw数据都是32位保持一致，方便后面使用解析。来源的bmp图片可能是24/32位两种，需要做一个判断，将RGB888格式的补全。

import cv2
import sys

def bmp_to_raw(filepath, outpath):
    """
    将 BMP 格式转换成 raw 格式数据，统一按照 32 位转raw，即 RGB888 和 RGB8888 格式图片转换后数据大小是一样的
    :param filepath: 待转换图片位置
    :param outpath: 转换后图片位置
    :return:
    """
    try:
        img = cv2.imread(filepath, cv2.IMREAD_UNCHANGED)
        channels = img.shape
    except AttributeError:
        print("图片加载失败，检查文件是否存在或是被占用")
        return -1
    except PermissionError:
        print("图片打开权限异常")
        return -1

    print("图片channels=", channels)

    if channels[2] == 4:
        # 如果是4通道的格式，直接转换输出即可
        try:
            img.tofile(outpath)
        except PermissionError:
            print("图片打开权限异常")
            return -1
        print("RGB8888 转换OK", outpath)

    elif channels[2] == 3:
        try:
            img.tofile('temp.raw')
        except PermissionError:
            print("图片打开权限异常")
            return -1
        # 3 通道的，缺失的数据手动给补上 0xff 填充
        fw = open(outpath, 'wb')
        with open('temp.raw', 'rb') as f:
            cont = f.read(1)
            fw.write(cont)
            i = 1;
            while len(cont) > 0:
                cont = f.read(1)
                if i % 4 == 3:
                    fw.write(b'\xff')
                    i += 2
                else:
                    i += 1
                fw.write(cont)

            f.close()
        fw.close()
        print("RGB888 转换OK", outpath)

if __name__ == '__main__':
    argLen = len(sys.argv)
    if argLen != 3:
        print("参数缺失错误。正确格式例如：\nmain.py test.bmp out.raw")
    else:
        filepath = sys.argv[1].strip()
        outpath = sys.argv[2].strip()

        print(filepath, outpath)
        if filepath.endswith('.bmp') and outpath.endswith('.raw'):
            bmp_to_raw(filepath, outpath)
        else:
            print("参数文件格式错误。正确格式例如：\nmain.py test.bmp out.raw")