OpenCV视觉分析之目标跟踪(8)目标跟踪函数CamShift()使用

  • 操作系统:ubuntu22.04
  • OpenCV版本:OpenCV4.9
  • IDE:Visual Studio Code
  • 编程语言:C++11

算法描述

找到物体的中心、大小和方向。

CamShift(Continuously Adaptive Mean Shift)是 OpenCV 中的一种目标跟踪算法,广泛应用于视频中对象的跟踪。CamShift 算法结合了 Mean Shift 算法和颜色直方图,通过迭代寻找目标区域的颜色分布中心,从而实现目标的跟踪。

基本原理

  • 颜色直方图:首先,根据目标区域的颜色信息构建颜色直方图。颜色直方图描述了目标区域中各颜色的分布情况。
  • Mean Shift:Mean Shift 算法通过迭代寻找目标区域的颜色分布中心,逐步调整目标区域的位置和大小。
  • 适应性调整:CamShift 算法不仅调整目标区域的位置,还根据目标区域的颜色分布调整其大小和方向,从而实现更精确的跟踪。

代码示例

cpp 复制代码
#include <iostream>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace std;

Rect selection;
bool selecting   = false;
bool trackObject = false;
Point origin;
Mat frame;  // 声明一个全局变量来存储当前帧

void onMouse( int event, int x, int y, int, void* )
{
    if ( selecting )
    {
        selection.x      = MIN( x, origin.x );
        selection.y      = MIN( y, origin.y );
        selection.width  = abs( x - origin.x );
        selection.height = abs( y - origin.y );
        selection &= Rect( 0, 0, frame.cols, frame.rows );  // 使用全局变量 frame
    }

    switch ( event )
    {
    case EVENT_LBUTTONDOWN:
        selecting = true;
        origin    = Point( x, y );
        selection = Rect( x, y, 0, 0 );
        break;
    case EVENT_LBUTTONUP:
        selecting = false;
        if ( selection.width > 0 && selection.height > 0 )
        {
            trackObject = true;
        }
        break;
    }
}

int main()
{
    VideoCapture cap(0);
    if ( !cap.isOpened() )
    {
        cerr << "Error opening video file" << endl;
        return -1;
    }

    namedWindow( "CamShift Tracker", WINDOW_NORMAL );
    setMouseCallback( "CamShift Tracker", onMouse );

    while ( true )
    {
        cap >> frame;
        if ( frame.empty() )
            break;

        Mat hsv_frame, hue_frame, mask, backproj;
        cvtColor( frame, hsv_frame, COLOR_BGR2HSV );

        if ( trackObject )
        {
            // 提取 H(色调)通道
            vector< Mat > hsv_channels;
            split( hsv_frame, hsv_channels );
            Mat hue_channel = hsv_channels[ 0 ];

            // 计算颜色直方图
            Mat roi = hue_channel( selection );
            Mat hsv_hist;
            int histSize[]        = { 32 };
            float hranges[]       = { 0, 180 };
            const float* ranges[] = { hranges };
            calcHist( &roi, 1, 0, Mat(), hsv_hist, 1, histSize, ranges, true, false );
            normalize( hsv_hist, hsv_hist, 0, 255, NORM_MINMAX );

            // 计算反投影
            calcBackProject( &hue_channel, 1, 0, hsv_hist, backproj, ranges, 1, true );

            // 使用 CamShift 进行跟踪
            TermCriteria term_crit( TermCriteria::COUNT | TermCriteria::EPS, 10, 1 );
            RotatedRect track_box = CamShift( backproj, selection, term_crit );

            // 绘制跟踪结果
            ellipse( frame, track_box, Scalar( 0, 255, 0 ), 2 );
        }

        // 绘制选择区域
        if ( selecting )
        {
            rectangle( frame, selection, Scalar( 255, 0, 0 ), 2 );
        }

        imshow( "CamShift Tracker", frame );

        char c = waitKey( 30 );
        if ( c == 27 )
            break;  // 按下 ESC 键退出
    }

    cap.release();
    destroyAllWindows();

    return 0;
}

运行结果

相关推荐
学习前端的小z7 分钟前
【GPTs】Front-end Expert:助力前端开发的智能工具
人工智能·gpt·chatgpt·aigc
使者大牙19 分钟前
【LLM学习笔记】第四篇:模型压缩方法——量化、剪枝、蒸馏、分解
人工智能·深度学习·算法·机器学习
Matlab程序猿小助手20 分钟前
【MATLAB源码-第222期】基于matlab的改进蚁群算法三维栅格地图路径规划,加入精英蚁群策略。包括起点终点,障碍物,着火点,楼梯。
开发语言·人工智能·算法·matlab·机器人·无人机
岛屿旅人22 分钟前
2025-2026财年美国CISA国际战略规划(下)
网络·人工智能·安全·web安全·网络安全
卧式纯绿26 分钟前
自动驾驶3D目标检测综述(三)
人工智能·python·深度学习·目标检测·3d·cnn·自动驾驶
不朽の燃梦1 小时前
OpenCV图像视频分析(1)
opencv·计算机视觉·音视频
ZOMI酱1 小时前
【AI系统】GPU 架构回顾(从2018年-2024年)
人工智能·架构
土豆炒马铃薯。2 小时前
【深度学习】Pytorch 1.x 安装命令
linux·人工智能·pytorch·深度学习·ubuntu·centos
阿_旭2 小时前
【超全】目标检测模型分类对比与综述:单阶段、双阶段、有无锚点、DETR、旋转框
人工智能·深度学习·目标检测·分类
研一计算机小白一枚2 小时前
Which Tasks Should Be Learned Together in Multi-task Learning? 译文
人工智能·python·学习·机器学习