Hisilicon AI chip (hi35xx): image JPG to BGR upgrade

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

  I improved the code of this brother above , Small improvements , Under backup , Lest you lose

        Mainly this ,self.img_size_w = 220  # save bgr size
         self.img_size_h = 416  # save bgr size

        The original one-dimensional data , I can distinguish the width and height here

        In addition, I added test_Hi_bgr1 You can read the width and height from the file name

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

        Is a simple string processing , If you want to use , Change it according to your file name , My file format is

XX-100-200.bgr, And my path E:/GraduateE_Competition/data/fu There is no ‘’ - ‘’ Of , All can put 100 200 Take it out

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  Parameter setting 
        self.imgpath_jpg = r"E:\\GraduateE_Competition\\jpg2bgr\\dog4.jpg"
        self.saveimg_bgr = r"E:\\GraduateE_Competition\\jpg2bgr\\dog4.bgr"
        # testbgr  converbgr=false  Parameter setting 
        self.jpeg_path = r"E:\\GraduateE-Competition\\jpg2bgr\\HumanWithDog.jpg"  #  Just to get the height and width of the picture 
        #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"

    """ Hayes nnie The model needs input bgr  Format picture , This python Script can put jpg Format of the picture into .bgr Format picture """

    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")

    """ see bgr The contents of the file are displayed as pictures """

    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")