python+opencv选出视频中一帧再利用鼠标回调实现图像上画矩形框

Posted 2020-09-21

　　最近因为要实现模板匹配，需要在视频中选中一个目标，然后框出（即作为模板），对其利用模板匹配的方法进行检测。于是需要首先选出视频中的一帧，但是在利用摄像头读视频的过程中我唯一能想到的方法就是：

1.在视频播放的过程中，当想选中哪幅图来标记目标时，按下暂停键。

2.框出你想要检测的目标。

那要怎么实现呢？

首先是暂停怎么实现呢？介于是摄像头直播，我们先将那副图像复制并在另一个窗口显示。然后关掉摄像头，就完成了暂停。后面标记完目标再打开摄像头就行了。那具体怎么实现呢...

import cv2

drawing = False
ix, iy = -1, -1
tempFlag = False
def draw_circle(event, x, y, flags, param):
    global ix, iy, drawing, mode, cap, template, tempFlag
    if event == cv2.EVENT_LBUTTONDOWN:
        tempFlag = True
        drawing = True
        ix, iy = x, y                        #按下鼠标左键，用全局变量ix,iy记录下当前坐标点
    elif event == cv2.EVENT_LBUTTONUP:
        if drawing == True:
            drawing = False                  #鼠标左键抬起，画出矩形框
            cv2.rectangle(frame, (ix, iy), (x, y), (0, 255, 0), 1)
            template = frame[iy:y, ix:x, :]  #截取框中的目标图像
            cap = cv2.VideoCapture(-1)       #打开摄像头
            cv2.imshow(‘img‘, frame)         #显示画框后的图像

cv2.namedWindow(‘img‘)
cv2.setMouseCallback(‘img‘, draw_circle)
cap = cv2.VideoCapture(-1)
while (True):
    ret, frame = cap.read()
    cv2.imshow(‘Vedio‘, frame)
    if tempFlag == True:         #如果框出了目标，显示该目标
        cv2.imshow(‘temp‘, template)
    k = cv2.waitKey(33)
    if k == 27:                  #退出视频
        break
    elif k == 32:                #如果按下空格键
        while(1):
            cap.release()        #关掉摄像头
            imgCOPY = frame      #显示关闭摄像头前最后一张图像
            cv2.imshow(‘img‘, frame)
            k = cv2.waitKey(0)   #等待调用鼠标回调函数框出目标
            if k == 32:          #框完目标，再次按下空格键，摄像头捕捉的画面重新播放
                break

结果如图

对于在图像中框出目标，其实我先看了网上的代码，比如http://blog.csdn.net/szfhy/article/details/51087301

我画框后的图片是这样的

内心是崩溃的，宝宝就只想要一个矩形框嘛~

本来我是想在画矩形框时，有一个矩形框随鼠标拖动大小随之改变，且只有一个矩形框，鼠标抬起时，矩形框确定。暂时不知怎么实现，有知道的可以交流一下，嘿嘿~