海思AI芯片(Hi35XX): 图像jpg转.bgr升级版

https://blog.csdn.net/smf0504/article/details/106068342

 我把上面这个老哥的代码改进了一下，小改进，备份下，免得丢了

        主要是这个，self.img_size_w = 220  # save bgr size
         self.img_size_h = 416  # save bgr size

        原来的只有一维数据，我这里区分宽高了

        另外就是我加了 test_Hi_bgr1 可以从文件名读宽高

        self.img_size_w = int(dir.split("-")[1]) 
        self.img_size_h = int(dir.split("-")[2])  

        就是一个简单的字符串处理，你要用的话，要根据自己文件名改改，我的文件格式是

XX-100-200.bgr, 还有我的路径E:/GraduateE_Competition/data/fu 中是没有 ‘’ - ‘’ 的，所有能把100 200 拿出来

import cv2

class JPG2BGR_Solver(object):
    def __init__(self):
        self.img_size_w = 220  # save bgr size
        self.img_size_h = 416  # save bgr size
        # jpj2bgr  converbgr=true 参数设置
        self.imgpath_jpg = r"E:\\GraduateE_Competition\\jpg2bgr\\dog4.jpg"
        self.saveimg_bgr = r"E:\\GraduateE_Competition\\jpg2bgr\\dog4.bgr"
        # testbgr  converbgr=false 参数设置
        self.jpeg_path = r"E:\\GraduateE-Competition\\jpg2bgr\\HumanWithDog.jpg"  # 只是为了获取图片高宽
        #self.path = "E:\\GraduateE-Competition\\jpg2bgr\\HumanWithDog_transferred.bgr"
        self.path = "E:\\GraduateE-Competition\\didi\\crop-65-308-304089172.bgr"
        #self.path = "E:\\GraduateE-Competition\\jpg2bgr\\cropped\\dog_bike_car_416x416.bgr"

    """海思nnie模型需要输入bgr 格式的图片，这个python脚本可以把jpg格式的图片转换成.bgr格式的图片"""

    def jpg2bgr(self):
        save_img_size_h = self.img_size_h
        save_img_size_w = self.img_size_w
        imgpath_jpg = self.imgpath_jpg
        img = cv2.imread(imgpath_jpg)

        if img is None:
            print("img is none")
        else:
            img = cv2.resize(img, (save_img_size_w, save_img_size_h))
            (B, G, R) = cv2.split(img)
            with open(self.saveimg_bgr, 'wb') as fp:
                for i in range(save_img_size_h):
                    for j in range(save_img_size_w):
                        fp.write(B[i, j])
                        # print(B[i, j])
                for i in range(save_img_size_h):
                    for j in range(save_img_size_w):
                        fp.write(G[i, j])
                for i in range(save_img_size_h):
                    for j in range(save_img_size_w):
                        fp.write(R[i, j])

            print("save success")

    """查看bgr文件内容并显示为图片"""

    def test_Hi_bgr(self):

        jpeg_path = self.jpeg_path
        path = self.path
        imgsize_h = self.img_size_h
        imgsize_w = self.img_size_w
        f = open(path, 'rb')
        src = cv2.imread(jpeg_path)
        src = cv2.resize(src, (imgsize_w, imgsize_h))
        print(src.shape)
        h = src.shape[0]
        w = src.shape[1]
        c = src.shape[2]
        print(f.name)
        (B, G, R) = cv2.split(src)

        data = f.read(imgsize_h * imgsize_w * 3)
        for j in range(imgsize_h):
            for i in range(imgsize_w):
                B[j, i] = data[j * imgsize_w + i]
                G[j, i] = data[j * imgsize_w + i + imgsize_w * imgsize_h]
                R[j, i] = data[j * imgsize_w + i + imgsize_w * imgsize_h * 2]

        newimg = cv2.merge([B, G, R])

        cv2.imshow("new", newimg)

        f.close()
        cv2.waitKey(0)
    def test_Hi_bgr1(self,dir):
        self.path = dir
        jpeg_path = self.jpeg_path
        path = self.path
        self.img_size_w = int(dir.split("-")[1])  # save bgr size
        self.img_size_h = int(dir.split("-")[2])   # save bgr size
        imgsize_h = self.img_size_h
        imgsize_w = self.img_size_w
        f = open(path, 'rb')
        src = cv2.imread(jpeg_path)
        src = cv2.resize(src, (imgsize_w, imgsize_h))
        print(src.shape)
        h = src.shape[0]
        w = src.shape[1]
        c = src.shape[2]
        print(f.name)
        (B, G, R) = cv2.split(src)

        data = f.read(imgsize_h * imgsize_w * 3)
        for j in range(imgsize_h):
            for i in range(imgsize_w):
                B[j, i] = data[j * imgsize_w + i]
                G[j, i] = data[j * imgsize_w + i + imgsize_w * imgsize_h]
                R[j, i] = data[j * imgsize_w + i + imgsize_w * imgsize_h * 2]

        newimg = cv2.merge([B, G, R])

        # cv2.imshow("new", newimg)
        cv2.imwrite(dir+".jpg", newimg)
        f.close()
        # cv2.waitKey(0)


if __name__ == '__main__':
    path_name='E:/GraduateE_Competition/data/fu'

    import os
    i = 0
    for item in os.listdir(path_name):

        converbgr = False
        solverObj = JPG2BGR_Solver()
        if (converbgr == True):
            solverObj.jpg2bgr()
        else:
            solverObj.test_Hi_bgr1(path_name+"/"+item)

    print("batch rename finished\n")