對視訊中車輛車牌進行識別並圈出

import hyperlpr  # 匯入hyperlpr庫，處理車牌識別
import cv2  # 匯入CV2庫，處理圖片或視訊
import numpy as np
from PIL import ImageFont, ImageDraw
from PIL import Image


def cv2AddChineseText(img, text, position, textColor=(0, 255, 0), textSize=30):
    if (isinstance(img, np.ndarray)):  # 判斷是否OpenCV圖片型別
        img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    # 建立一個可以在給定影象上繪圖的物件
    draw = ImageDraw.Draw(img)
    # 字型的格式
    fontStyle = ImageFont.truetype(
        "simsun.ttc", textSize, encoding="utf-8")
    # 繪製文字
    draw.text(position, text, textColor, font=fontStyle)
    # 轉換回OpenCV格式
    return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)


def main():
    mp4 = cv2.VideoCapture("2.mp4")  # 讀出視訊的路徑
    while (mp4.isOpened()):  # 使用mp4.isOpened()函數來判斷視訊是否正常開啟
        ret, frame = mp4.read()  # 讀取視訊，第一個引數ret 為True 或者False,代表有沒有讀取到圖片，第二個引數frame表示擷取到一幀的圖片
        if ret == True:  # 判斷視訊有沒有結束
            gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)  # 將每一幀圖片轉為灰色以便接下來處理
            k = hyperlpr.HyperLPR_plate_recognition(frame)  # 使用HyperLPR_plate_recognition函數識別圖片內車牌號
            print(type(k))
            if k:
                d = []
                a = k[0][0]
                b = k[0][1]
                c = k[0][2]
                print(a)
                print(b)
                b = str(b)
                print(c)
                d.append(c)
                f = np.array(d, dtype=int)
                print(type(d))
                for (x, y, w, h) in d:
                    cv2.rectangle(frame, (x, y), (w, h), (0, 0, 255), 2)
                    frame = cv2AddChineseText(frame, a, (x, y), (255, 0, 0), 30)
                    cv2.putText(frame, b, (x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
            cv2.imshow('frame', frame)
            # 視訊正常開啟並把識別的車牌號寫入jianVideo.txt內
            if cv2.waitKey(1) & 0xFF == ord('q'):  # 類似中斷播放的按鍵，按q跳出迴圈終止播放
                break
        else:  # 如果視訊結束正常跳出迴圈終止播放
            break
    mp4.release()  # 釋放視訊
    cv2.destroyAllWindows()  # 將建立的所有的視窗銷燬


if __name__ == '__main__':
    main()