Python Opencv实践 - 视频目标追踪CamShift

CamShift是MeanShift的改进，能够动态自适应跟踪目标大小，而不是一个固定窗口。在opencv中使用CamShift进行目标追踪的方法和MeanShift差不多，只需要替换meanShift方法并且使用cv.polylines绘制出结果多边形窗口即可。

参考资料：

Python Opencv实践 - 视频目标追踪MeanShift_亦枫Leonlew的博客-CSDN博客

python opencv入门 Meanshift 和 Camshift 算法（40）_meanshift源码opencv_红鲤鱼遇绿鲤鱼的博客-CSDN博客

import cv2 as cv
import numpy as np
import matplotlib.pyplot as plt

#打开视频
video = cv.VideoCapture("../SampleVideos/milo.mp4")
#读取第一帧图像
ret,frame = video.read()
plt.figure(figsize=(16,16))
plt.imshow(frame[:,:,::-1])
cv.imwrite("tmp.jpg", frame)


#设置跟踪目标的位置(行，高，列，宽)
row,height,col,width = 670,80,360,120
obj_track_window = (col, row, width, height)
#设置ROI区域
roi = frame[row:row+height,col:col+width]
plt.imshow(roi[:,:,::-1])

#计算ROI的直方图
#1. 转换到HSV色彩空间
roi_hsv = cv.cvtColor(roi, cv.COLOR_BGR2HSV)
#2. 计算ROI区域直方图
roi_hist = cv.calcHist([roi_hsv],[0],None,[180],[0,180])
#   直方图归一化
cv.normalize(roi_hist, roi_hist, 0, 255, cv.NORM_MINMAX)

#进行目标追踪
#1. 设置窗口搜索终止条件：最大迭代次数，窗口中心漂移最小值
#参考资料：https://blog.csdn.net/csg3140100993/article/details/102613843
term_crit = (cv.TERM_CRITERIA_EPS|cv.TERM_CRITERIA_COUNT, 10, 1)

#2. 处理每一帧
while (True):
    ret,frame = video.read()
    if ret == False:
        break;
    #计算直方图反向投影
    #cv.calcBackProject(image, channels, hist, ranges, scale[, dst])
    #image:输入图像（HSV），要用中括号[]括起来
    #channels:用于计算反向投影的通道列表，通道数必须与直方图的维度匹配
    #hist:模板图像的直方图
    #ranges：直方图中每个维度bin的取值范围
    #scale：缩放因子，一般为1
    #参考资料：https://www.jianshu.com/p/d0a2c70cf99e
    hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV)
    target = cv.calcBackProject([hsv],[0],roi_hist,[0,180],1)
    #使用MeanShit追踪
    #cv.CamShift(probImage, window, criteria)
    #probImage：ROI区域，即目标直方图的反向投影
    #window：初始搜索窗口，就是定义ROI的rect
    #criteria：确定窗口搜索停止的准则，主要有迭代次数达到设置的最大值，窗口中心的漂移值大于某个设定的限值等
    #参考资料：https://blog.csdn.net/tengfei461807914/article/details/80412482
    ret,obj_track_window = cv.CamShift(target, obj_track_window, term_crit)
    #绘制追踪目标多边形
    points = cv.boxPoints(ret)
    points = np.intp(points)
    img_track = cv.polylines(frame, [points], True, (0,255,0), 2)
    cv.imshow('MeanShift Demo', img_track)
    if cv.waitKey(30) & 0xFF == ord('q'):
        break;

video.release()
cv.destroyAllWindows()

Python Opencv实践 - 视频目标追踪CamShift

猜你喜欢