python+opencv选出视频中一帧再利用鼠标回调实现图像上画矩形框
最近因为要实现模板匹配,需要在视频中选中一个目标,然后框出(即作为模板),对其利用模板匹配的方法进行检测。于是需要首先选出视频中的一帧,但是在利用摄像头读视频的过程中我唯一能想到的方法就是:
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
我画框后的图片是这样的
内心是崩溃的,宝宝就只想要一个矩形框嘛~
本来我是想在画矩形框时,有一个矩形框随鼠标拖动大小随之改变,且只有一个矩形框,鼠标抬起时,矩形框确定。暂时不知怎么实现,有知道的可以交流一下,嘿嘿~