简单模板匹配实现跟踪

环境python2+opencv2

效果：打开摄像头，鼠标选择目标，自动跟踪

优缺点：可以跟踪，但会突然跟丢。算法问题。

# -*- coding: utf-8 -*-
import cv2
import numpy as np

tem=[0,0,0,0]
tem_im=[]
ix,iy=-1,-1
ox,oy=-1,-1
#0 nothing 1 template choose 2 show the destination
mode=0

def draw_rec(event,x,y,flags,param):
global tem_im,tem,ix,iy,ox,oy,mode

if event==cv2.EVENT_LBUTTONDOWN:
ix=x
iy=y
mode=1
'''
if event==cv2.EVENT_MOUSEMOVE and flags==cv2.EVENT_FLAG_LBUTTON:
cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),1)
'''
ox=x
oy=y
if event==cv2.EVENT_LBUTTONUP:
ox=x
oy=y
mode=2
tem=[ix,iy,ox,oy]
tem_im=img[tem[1]:tem[3],tem[0]:tem[2]].copy()

cap=cv2.VideoCapture(0)
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_rec)
while(1):
success,img=cap.read()
if mode==1:
cv2.rectangle(img,(ix,iy),(ox,oy),(0,255,0),1)
if mode==2:
res = cv2.matchTemplate(img,tem_im,cv2.TM_SQDIFF)
min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)
rect_p=(min_loc[0]+tem[2]-tem[0],min_loc[1]+tem[3]-tem[1])
cv2.rectangle(img,min_loc,rect_p,(0,255,0),1)

cv2.imshow('image',img)
if cv2.waitKey(20)&0xFF==27:
break
cv2.destroyAllWindows()