使用cv2.selectROI获得视频第一帧图像中某部分ROI框图的坐标信息

原创于 2020-11-29 20:58:49 发布 · 1.4k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#计算机视觉 #python

该篇博客介绍了如何在Python中利用cv2库的selectROI函数，从视频第一帧中获取感兴趣区域（ROI）的坐标信息。在没有图形界面的情况下，通过保存第一帧为图片并运行代码，可以预先设定ROI，用于后续的目标追踪任务。

使用cv2.selectROI得出视频第一帧图像中某部分ROI框图的坐标信息

ROI（region of interest），即感兴趣区域。在做目标追踪时在目标出现的第一帧图像中采用鼠标截取的方式，可以确定跟踪目标。自己在做试验时，由于没有图形界面，无法在视频中画框图，于是在读取的视频第一帧中，先使用cv2.selectROI得到图像的坐标信息，输入程序中，代替用鼠标截取的方式得出ROI值。

Python代码如下：


import cv2

def get_frames():

    video_name = "../bag.avi"

    cap = cv2.VideoCapture(video_name)
    ret, frame = cap.read()
    if ret:
        first_frame = "../bag1.jpg"
        cv2.imwrite(first_frame, frame)
        init_rect = cv2.selectROI(frame, False, False)
        print(init_rect)

if __name__ == "__main__":
    get_frames()