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;
}

运行结果

相关推荐
DO_Community3 小时前
普通服务器都能跑:深入了解 Qwen3-Next-80B-A3B-Instruct
人工智能·开源·llm·大语言模型·qwen
WWZZ20253 小时前
快速上手大模型:机器学习3(多元线性回归及梯度、向量化、正规方程)
人工智能·算法·机器学习·机器人·slam·具身感知
deephub3 小时前
深入BERT内核:用数学解密掩码语言模型的工作原理
人工智能·深度学习·语言模型·bert·transformer
PKNLP3 小时前
BERT系列模型
人工智能·深度学习·bert
兰亭妙微4 小时前
ui设计公司审美积累 | 金融人工智能与用户体验 用户界面仪表盘设计
人工智能·金融·ux
AKAMAI4 小时前
安全风暴的绝地反击 :从告警地狱到智能防护
运维·人工智能·云计算
岁月宁静4 小时前
深度定制:在 Vue 3.5 应用中集成流式 AI 写作助手的实践
前端·vue.js·人工智能
galaxylove4 小时前
Gartner发布数据安全态势管理市场指南:将功能扩展到AI的特定数据安全保护是DSPM发展方向
大数据·人工智能
格林威5 小时前
偏振相机在半导体制造的领域的应用
人工智能·深度学习·数码相机·计算机视觉·视觉检测·制造
晓枫-迷麟6 小时前
【文献阅读】当代MOF与机器学习
人工智能·机器学习