Python知识点-例子1-显示YUV420SP数据

import cv2
import numpy as np
import argparse

class ShowYUV:
    def __init__(self, filename, size):
        self.height, self.width = size
        self.frame_len = int(self.width * self.height * 3 / 2)
        self.f = open(filename, 'rb')
        self.shape = (int(self.height*1.5), self.width)

    def read_raw(self):
        try:
            raw = self.f.read(self.frame_len)
            yuv = np.frombuffer(raw, dtype=np.uint8)
            yuv = yuv.reshape(self.shape)
        except Exception as e:
            print(str(e))
            return False, None
        return True, yuv

    def read(self):
        ret, yuv = self.read_raw()
        if not ret:
            return ret, yuv
        bgr = cv2.cvtColor(yuv, cv2.COLOR_YUV420sp2BGR)
        return ret, bgr


if __name__ == "__main__":
    # construct the argument parse and parse the arguments
    ap = argparse.ArgumentParser()
    ap.add_argument("-i", "--image", required = True, help = "Path to the image")
    ap.add_argument("-c", "--cols", required = True, help = "Path to cols of the image")
    ap.add_argument("-r", "--rows", required = True, help = "Path to rows of the image")
    args = vars(ap.parse_args())

    filename = args["image"]
    size = (int(args["rows"]), int(args["cols"]))
    
    print("{} {} {}".format(filename, size[0], size[1]))
    
    cap = ShowYUV(filename, size)
    _, frame = cap.read()
    cv2.imshow("frame", frame)
    cv2.waitKey(0)

最后的话:

我是一个工作10年的程序员,工作中经常会遇到需要查一些关键技术,但是很多技术名词的介绍都写的很繁琐,为什么没有一个简单的/5分钟能说清楚的博客呢. 我打算有空就写写这种风格的指南文档.CSDN上搜蓝色的杯子, 没事多留言,指出我写的不对的地方,写的排版风格之类的问题,让我们一起爱智求真吧.wisdomfriend@126.com是我的邮箱,也可以给我邮箱留言.