raw转png python实现

import cv2
import numpy as np
import os

path = "F:\IMAGE\small_orial_color".replace('\\', '/')
save_path = "F:\IMAGE\small_orial_ir_png".replace('\\', '/')
file_names = os.listdir(path)

for name in file_names:
    if "raw" in name:
        print(path + "/" + name)
        if "orial_rgb" in name:
            #将raw格式的rgb转成png
            rawImg = np.fromfile(path + "/" + name, dtype=np.uint8)
            try:
                print('rawImg.shape', rawImg.shape)
                rawImg1 = rawImg[:635904].reshape(864, 736)
                rawImg2 = rawImg[635904:1271808].reshape(864, 736)
                rawImg3 = rawImg[1271808:].reshape(864, 736)
                rawImg = cv2.merge([rawImg1, rawImg2, rawImg3])
                h, w, d = rawImg.shape
                print(h, w, d)
                cv2.imshow("img", rawImg)
                cv2.waitKey(5)
                newName = name.replace("raw", "png")
                cv2.imwrite(save_path + "/" + newName, rawImg)
            except:
                continue
        elif "orial_depth" in name:
            rawImg = np.fromfile(path + "/" + name, dtype=np.uint8)
            try:
                rawImg = rawImg.reshape(112, 112)
                h, w = rawImg.shape
                print(h, w)
                cv2.imshow("img", rawImg)
                cv2.waitKey(5)
                newName = name.replace("raw", "png")
                cv2.imwrite(save_path + "/" + newName, rawImg)
            except:
                continue
        elif "orial_ir" in name:
            rawImg = np.fromfile(path + "/" + name, dtype=np.uint16)
            try:
                rawImg = rawImg.reshape(112, 112)
                h, w = rawImg.shape
                print(h, w)
                cv2.imshow("img", rawImg/4)
                cv2.waitKey(5)
                newName = name.replace("raw", "png")
                cv2.imwrite(save_path + "/" + newName, rawImg/4)
            except:
                continue