【Python】matplotlib替代cv画虚线矩形框

最新推荐文章于 2024-08-10 23:14:03 发布

RockWang.

最新推荐文章于 2024-08-10 23:14:03 发布

阅读量404

点赞数

分类专栏：程序语言文章标签： python matplotlib opencv

本文为博主原创文章，未经博主允许不得转载。

本文链接：https://blog.csdn.net/xi_shui/article/details/129580616

版权

程序语言专栏收录该内容

47 篇文章 6 订阅

订阅专栏

该代码读取一个视频文件，使用OpenCV在每一帧上绘制矩形，然后保存处理后的视频，并通过Matplotlib进行实时显示。矩形坐标(a,b)为(2,3)和(140,160)，并以蓝色实线描绘。处理的每一帧也会保存为图片文件。

摘要由CSDN通过智能技术生成

import cv2 as cv
import matplotlib.pyplot as plt
import matplotlib.patches as patches
plt.ion() #开启动态绘图
plt.figure() #建立图窗
def main():
    cap = cv.VideoCapture("./data/testvideo1.mp4")
    sz = (int(cap.get(cv.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv.CAP_PROP_FRAME_HEIGHT)))
    fourcc = cv.VideoWriter_fourcc('m', 'p', '4', 'v')
    video_write = cv.VideoWriter("./data/1.mp4", fourcc, 30, sz, True)
    frame_cnt = 1
    while True:
        ret, frame = cap.read()
        if not ret:
            break
        a = (2, 3)
        b = (140, 160)
        cv.rectangle(frame, a, b, (255, 0, 0), 2, )
        rect = patches.Rectangle(a, b[0]-a[0], b[1]-a[1], linewidth=1, edgecolor='black', facecolor='none', linestyle='dotted')
        cv.imshow('Demo', frame)
        cv.imwrite("./image/{}.jpg".format(frame_cnt), frame)
        frame_cnt = + 1
        cv.waitKey(100)  # 显示 1000 ms 即 1s 后消失
        plt.imshow(frame[:, :, ::-1])
        currentAxis = plt.gca()
        currentAxis.add_patch(rect)
        plt.pause(0.01)
if __name__ == "__main__":
    main()