Python OpenCV目标跟踪，可在视频播放过程中选择标定框

Python OpenCV目标跟踪，可在视频播放过程中选择标定框

这里用到的OpenCV最好是安装opencv-contrib-python，用pip install opencv-contrib-python安装即可。
主要实现的功能如下：（这里本人只是为了做demo，OpenCV自带的目标跟踪效果并不是很好）

• 在视频播放过程中选择要跟踪的目标（按键盘上的“0”键选择目标）
• 可取消跟踪目标（按键盘上的“1”键取消跟踪目标）
• 改变跟踪框的颜色并显示车牌信息（按“2”，我这里是改为了绿色，可显示汉字，这里是把车牌的信息写成固定的，为了demo）
• 显示车牌的信息，
• 取消跟踪之后，可重新按“0”键选择跟踪目标

代码主要是为了备份以后使用，不一定满足大家的需要，谨慎运行

import cv2
import sys
import numpy
from PIL import Image, ImageDraw, ImageFont


def cv2ImgAddText(img, text, left, top, textColor=(255, 0, 0), textSize=20):
    if isinstance(img, numpy.ndarray):  # 判断是否OpenCV图片类型
        img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    # 创建一个可以在给定图像上绘图的对象
    draw = ImageDraw.Draw(img)
    # 字体的格式
    fontStyle = ImageFont.truetype(
        "font/simsun.ttc", textSize, encoding="utf-8")
    # 绘制文本
    draw.text((left, top), text, textColor, font=fontStyle)
    # 转换回OpenCV格式
    return cv2.cvtColor(numpy.asarray(img), cv2.COLOR_RGB2BGR)


if __name__ == '__main__':
    # Read video
    video = cv2.VideoCapture("车牌2.mp4")
    # 保存视频
    fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
    sz = (int(video.get(cv2.CAP_PROP_FRAME_WIDTH)), int(video.get(cv2.CAP_PROP_FRAME_HEIGHT)))
    fps = 30
    out = cv2.VideoWriter()
    out.open('output.mp4', fourcc, fps, sz, True)
    # Exit if video not opened.
    if not video.isOpened():
        print("Could not open video")
        sys.exit()
    # video.set(cv2.CAP_PROP_POS_FRAMES, 5600)  # 跳帧
    ok, frame = video.read()
    if not ok:
        print('Cannot read video file')
        sys.exit()
    key0 = 0  # 标记“Z"是否被按下, 偶数为不追踪， 奇数为追踪
    key2 = False  # 默认警告
    key3 = False
    while True:
        # Read a new frame
        k = cv2.waitKey(1) & 0xff
        ok, frame = video.read()
        if not ok:
            break
        if k == 48:  # 如果按下了“0”键，则重新选择bbox
            key0 += 1
            key2 = False  # 默认警告
            tracker = cv2.TrackerKCF_create()
            bbox = cv2.selectROI(frame, False)
            ok = tracker.init(frame, bbox)
            cv2.destroyAllWindows()
        if k == 49:  # 如果按下了“1”键，则取消bbox
            key0 += 1
        if k == 50:  # 如果按下了“2”键，则改变框的颜色
            key2 = True
        if k == 51:  # 如果按下了“3”键，则显示车牌信息
            key3 = True
        if key0 % 2 == 1:
            ok, bbox = tracker.update(frame)

        # Draw bounding box
        if ok and key0 % 2 == 1:
            # Tracking success
            p1 = (int(bbox[0]), int(bbox[1]))
            p2 = (int(bbox[0] + bbox[2]), int(bbox[1] + bbox[3]))
            if key2:
                cv2.rectangle(frame, p1, p2, (255, 0, 0), 2, 1)
                # cv2.putText(frame, " ", (int(bbox[0]), int(bbox[1]) - 20), cv2.FONT_HERSHEY_SIMPLEX,
                #             0.75, (0, 255, 0), 2)
                frame = cv2ImgAddText(frame, "鲁A 6JIN3", int(bbox[0]), int(bbox[1]) - 25, (0, 0, 255), 22)
            else:
                cv2.rectangle(frame, p1, p2, (0, 0, 255), 2, 1)
                cv2.putText(frame, "Press Line", (int(bbox[0]), int(bbox[1]) - 20),
                            cv2.FONT_HERSHEY_SIMPLEX,
                            0.75, (0, 0, 255), 2)

        # Display tracker type on frame
        cv2.putText(frame, "KFC Tracker", (100, 20), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (255, 0, 0), 2)
        # Display result
        cv2.imshow("Tracking", frame)
        out.write(frame)
        if k == 27:  # Exit if ESC pressed
            break

    video.release()
    out.release()
    cv2.destroyAllWindows()

视频不好传，传点截图吧