dfa6f460c00db11b040392fa7ed3bf76.png

71a5733d45869836cbe248d558d07f83.png

d8e4d133c37def60060cf4951c646a1a.png

该代码主要实现的功能是视频中光流的计算与可视化。首先使用OpenCV的VideoCapture对象打开视频文件,并每帧提取视频帧。接着,对连续两帧图像使用Dense Optical Flow,这里使用了DIS (Dense Inverse Search)算法来计算它们之间的光流向量。计算得到的光流向量分解为x和y分量,并转换为极坐标表示的大小和角度。这些信息通过归一化并映射到HSV色彩空间来对光流进行可视化,最终以颜色编码的方式显示出来。该过程包括了读取视频、处理每一帧并显示结果,直到遍历完所有帧或用户中断为止。

#include "opencv2/core/utility.hpp" // 包含OpenCV核心功能支持的头文件
#include "opencv2/highgui.hpp"      // 包含OpenCV GUI 和图像/视频读写的头文件
#include "opencv2/imgproc.hpp"      // 包含OpenCV图像处理的头文件
#include "opencv2/videoio.hpp"      // 包含OpenCV视频读写的头文件
#include "opencv2/video.hpp"        // 包含OpenCV视频分析的头文件


using namespace std; // 使用标准命名空间
using namespace cv;  // 使用OpenCV命名空间


int main(int argc, char **argv)
{
    // 解析命令行参数
    CommandLineParser parser(argc, argv, "{ @video  | vtest.avi  | use video as input }"); 
    // 获取视频文件名,如果没找到,则保留输入的字符串
    string filename = samples::findFileOrKeep(parser.get<string>("@video")); 


    VideoCapture cap; // 创建视频捕获对象
    cap.open(filename); // 打开视频文件


    // 如果无法打开视频文件,则打印错误信息并退出
    if(!cap.isOpened())
    {
        printf("ERROR: Cannot open file %s\n", filename.c_str());
        parser.printMessage();
        return -1;
    }


    // 定义图像矩阵
    Mat prevgray, gray, rgb, frame;
    // 光流图像和分量
    Mat flow, flow_uv[2];
    // 光流的大小与角度
    Mat mag, ang;
    // HSV色彩空间分量
    Mat hsv_split[3], hsv;
    // 接收键盘输入
    char ret;


    // 创建稠密光流计算对象(中等预设)
    Ptr<DenseOpticalFlow> algorithm = DISOpticalFlow::create(DISOpticalFlow::PRESET_MEDIUM);


    // 无限循环,直到视频读取完成或用户终止
    while(true)
    {
        cap >> frame; // 从视频中读取一帧
        // 如果读取到的帧为空,则跳出循环
        if (frame.empty())
            break;


        // 将图像由BGR色彩空间转换为灰度空间
        cvtColor(frame, gray, COLOR_BGR2GRAY);


        // 如果prevgray不为空,则计算当前帧和前一帧之间的光流
        if (!prevgray.empty())
        {
            algorithm->calc(prevgray, gray, flow); // 计算光流
            split(flow, flow_uv); // 分离光流的两个分量
            multiply(flow_uv[1], -1, flow_uv[1]); // 翻转y分量
            cartToPolar(flow_uv[0], flow_uv[1], mag, ang, true); // 将笛卡尔坐标转换为极坐标
            // 归一化光流的大小  
            normalize(mag, mag, 0, 1, NORM_MINMAX);
            // 是否设定HSV色彩空间的H、S、V分量
            hsv_split[0] = ang; // 角度字段
            hsv_split[1] = mag; // 光流大小字段
            hsv_split[2] = Mat::ones(ang.size(), ang.type()); // 光流可视化的V分量设为全1矩阵
            merge(hsv_split, 3, hsv); // 融合HSV的分量得到完整的HSV图像
            cvtColor(hsv, rgb, COLOR_HSV2BGR); // 将HSV色彩空间转化为BGR色彩空间
            imshow("flow", rgb); // 显示光流图像
            imshow("orig", frame); // 显示原始图像
        }


        // 捕获键盘输入,如果用户按下任何键,终止循环
        if ((ret = (char)waitKey(20)) > 0)
            break;
        std::swap(prevgray, gray); // 更新前一帧的图像
    }


    // 正常结束程序,返回0
    return 0;
}

9456ec7fbfb1e31033aaabe61befb535.png

std::swap(prevgray, gray);

e0aeab358450ea0462c484b7b701b276.png

algorithm->calc(prevgray, gray, flow);

2eddb6ef23154a2a1b2bea7d2598fd19.png

cartToPolar(flow_uv[0], flow_uv[1], mag, ang, true);

cc045071b7c9f53ea54e3b71439860bf.png

The End

GitHub 加速计划 / opencv31 / opencv
77.39 K
55.71 K
下载
OpenCV: 开源计算机视觉库
最近提交(Master分支:3 个月前 )
7be5181b Fixed KLEIDICV_SOURCE_PATH handling for external KleidiCV 2 天前
c3ca3f4f - 3 天前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐