说明:分两种方式处理第一帧图像,具体操作内容见注释case1,case2
# encoding: utf-8
import numpy as np
import cv2
current_pos = None
tl = None
br = None
#鼠标事件
def get_rect(im, title='img1'):
mouse_params = {'tl': None, 'br': None, 'current_pos': None,'released_once': False}
cv2.namedWindow(title)
cv2.moveWindow(title, 100, 0) #启动后初始窗口的位置
def onMouse(event, x, y, flags, param):
param['current_pos'] = (x, y)
if param['tl'] is not None and not (flags & cv2.EVENT_FLAG_LBUTTON):
param['released_once'] = True
if flags & cv2.EVENT_FLAG_LBUTTON:
if param['tl'] is None:
param['tl'] = param['current_pos']
elif param['released_once']:
param['br'] = param['current_pos']
# tl记录起始位置,br记录最终位置
cv2.setMouseCallback(title, onMouse, mouse_params)
cv2.imshow(title, im)
while mouse_params['br'] is None:
im_draw = np.copy(im)
if mouse_params['tl'] is not None:
cv2.rectangle(im_draw, mouse_params['tl'],
mouse_params['current_pos'], (255, 0, 0))
#此while循环条件为没有第二次按下,然后不断随鼠标移动画矩形
cv2.imshow(title, im_draw)
_ = cv2.waitKey(10)
cv2.destroyWindow(title)
tl = (min(mouse_params['tl'][0], mouse_params['br'][0]),
min(mouse_params['tl'][1], mouse_params['br'][1]))
br = (max(mouse_params['tl'][0], mouse_params['br'][0]),
max(mouse_params['tl'][1], mouse_params['br'][1]))
# t1表示矩形左上角的点,t2表示右下角的点
#print mouse_params['tl'],mouse_params['br']
return (tl, br)
def xuanze(frame):
a1,a2 = get_rect(frame, title='img1') #手动选框
r,h,c,w = a1[1],a2[1]-a1[1],a1[0],a2[0]-a1[0] #手动选框
track_window = (c,r,w,h)
# 设置所要跟踪的ROI
roi = frame[r:r+h, c:c+w]
hsv_roi = cv2.cvtColor(roi, cv2.COLOR_BGR2HSV)
mask = cv2.inRange(hsv_roi, np.array((0., 60.,32.)), np.array((180.,255.,255.)))
roi_hist = cv2.calcHist([hsv_roi],[0],mask,[180],[0,180])
cv2.normalize(roi_hist,roi_hist,0,255,cv2.NORM_MINMAX)
while(1):
ret ,frame = cap.read()
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
dst = cv2.calcBackProject([hsv],[0],roi_hist,[0,180],1)
term_crit = ( cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1 )
# apply meanshift to get the new location
ret, track_window = cv2.meanShift(dst, track_window, term_crit)
# Draw it on image
x,y,w,h = track_window
cv2.rectangle(frame, (x,y), (x+w,y+h), 255,2)
cv2.namedWindow('img2')
cv2.moveWindow('img2', 100, 0) #控制播放窗口位于屏幕中间
cv2.waitKey(30)
cv2.imshow('img2',frame)
cv2.destroyWindow('video')
k = cv2.waitKey(10) & 0xff
if k == 27:
break
cap = cv2.VideoCapture('yundong1.mp4')
cv2.namedWindow("video")
cv2.moveWindow("video", 100, 0)
#case1: 显示第一帧,询问用户是否发现目标(y/else)。
# y:提供新窗口供用户标出目标,随后跟踪。
#else:播放视频,用户发现目标,按空格暂停,标出目标后跟踪
# while (cap.isOpened()):
# ret, frame = cap.read()
# cv2.imshow('video', frame)
# cv2.waitKey(0)
# cv2.destroyWindow('video')
# c = raw_input("Do you find your target: ")
# if str(c)=='y':
# print 'Pless marked the target, the target will be tracked'
# frame=xuanze(frame)
# else:
# break
#case2:显示第一帧图像供用户查看,按任意键开始视频播放
#播放视频过程中,如果用户发现目标,按空格暂停,标出目标后跟踪
ret, frame = cap.read()
cv2.imshow('video', frame)
print 'This is the first frame of image, press any key to start playing video. '
print 'If you find the target, press the spacebar, and then marked the target, the target will be tracked'
cv2.waitKey(0)
while (cap.isOpened()):
ret, frame = cap.read()
cv2.waitKey(30)
cv2.imshow('video', frame)
if cv2.waitKey(1) == (32):
cv2.imshow('video', frame)
frame=xuanze(frame)
if cv2.waitKey(1) == (27):
break
cv2.waitKey(0)
cv2.destroyAllWindows()
cap.release()