【opencv】教程代码 —video(3) 视频背景剔除

bg_sub.cpp

这段代码的功能是把视频中的背景和前景分离,提取出前景的运动物体 。根据用户选择的不同的模式,可以选择基于MOG2或者基于KNN的方法 来进行背景减除。在处理每一帧图像的过程中,首先使用背景减除模型对图像帧进行处理,得到一个前景掩码,该掩码中包含了所有被认为是前景的像素。然后将处理后的掩码图像和原始帧同时显示出来 。通过这种方式,用户能够清楚看到前景移动物体被成功地从背景中分离出来,这个过程是实时的。

cpp 复制代码
/**
 * @file bg_sub.cpp
 * @brief 背景减除教程样例代码
 * @author Domenico D. Bloisi
 */


#include <iostream>  // 引入 C++ 的输入输出流库
#include <sstream>  // 引入字符串流,它是 iostream 的一部分,提供了方便的字符串输入和输出功能
#include <opencv2/imgcodecs.hpp>  // 引入 OpenCV 的图像编解码模块,用于图像的读取和保存
#include <opencv2/imgproc.hpp>  // 引入 OpenCV 的图像处理模块,包含很多图像处理的功能
#include <opencv2/videoio.hpp>  // 引入 OpenCV 的视频 IO 模块,用于视频保存和读取、以及本机摄像头的访问
#include <opencv2/highgui.hpp>  // 引入 OpenCV 的高级 GUI 模块,用于创建界面显示图像
#include <opencv2/video.hpp>  // 引入 OpenCV 的视频分析模块,包含了许多常见的视频分析算法,比如对象跟踪、运动估计等


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


const char* params
    = "{ help h         |           | Print usage }"
      "{ input          | vtest.avi | Path to a video or a sequence of image }"
      "{ algo           | MOG2      | Background subtraction method (KNN, MOG2) }";


int main(int argc, char* argv[])  
{
    // 命令行参数解析
    CommandLineParser parser(argc, argv, params);
    string about_message = "This program shows how to use background subtraction methods provided by "
                  " OpenCV. You can process both videos and images.\n";
    parser.about(about_message);


    if (parser.has("help"))   // 如果有帮助信息
    {
        parser.printMessage();  // 打印使用帮助信息
    }


    // 创建背景减除对象
    Ptr<BackgroundSubtractor> pBackSub;
     //根据输入确定使用 KNN 还是 MOG2 
    if (parser.get<String>("algo") == "MOG2")
        pBackSub = createBackgroundSubtractorMOG2();
    else
        pBackSub = createBackgroundSubtractorKNN();


    // 输入视频
    VideoCapture capture( samples::findFile(parser.get<String>("input")) );
    if (!capture.isOpened()){  
        cerr << "Unable to open: " << parser.get<String>("input") << endl;
        return 0;
    }
    Mat frame, fgMask;  // 定义图像矩阵,用于读取视频帧以及存储背景减除图像
    while (true) {
        capture >> frame;  // 读取一帧
        if (frame.empty())
            break;


        // 更新背景模型
        pBackSub->apply(frame, fgMask);
        
        // 在当前帧上显示帧编号
        rectangle(frame, cv::Point(10, 2), cv::Point(100,20), cv::Scalar(255,255,255), -1);
        // 在 frame 上画一个白色矩形,左上角坐标为(10, 2),右下角坐标为(100, 20),颜色为白色(即 (255,255,255))。
        
        stringstream ss;
        ss << capture.get(CAP_PROP_POS_FRAMES);
        // 用 stringstream 将视频的当前帧数转换为字符串。
        
        string frameNumberString = ss.str();
        // 将帧数从 stringstream 转换为 string。
        
        putText(frame, frameNumberString.c_str(), cv::Point(15, 15), FONT_HERSHEY_SIMPLEX, 0.5 , cv::Scalar(0,0,0));
        // 在 frame 上的(15, 15)位置用黑色字体输出当前的帧数。
        // 显示当前帧和背景掩码
        imshow("Frame", frame);
        imshow("FG Mask", fgMask);
       
        int keyboard = waitKey(30);  // 获取键盘输入
        if (keyboard == 'q' || keyboard == 27)
            break;
    }
    return 0;
}
相关推荐
安静读书2 小时前
Python解析视频FPS(帧率)、分辨率信息
python·opencv·音视频
小陈phd2 小时前
OpenCV从入门到精通实战(九)——基于dlib的疲劳监测 ear计算
人工智能·opencv·计算机视觉
佑华硬盘拷贝机2 小时前
音频档案批量拷贝:专业SD拷贝机解决方案
音视频
EasyNVR2 小时前
NVR管理平台EasyNVR多个NVR同时管理:全方位安防监控视频融合云平台方案
安全·音视频·监控·视频监控
Guofu_Liao3 小时前
大语言模型---LoRA简介;LoRA的优势;LoRA训练步骤;总结
人工智能·语言模型·自然语言处理·矩阵·llama
ZHOU_WUYI7 小时前
3.langchain中的prompt模板 (few shot examples in chat models)
人工智能·langchain·prompt
如若1237 小时前
主要用于图像的颜色提取、替换以及区域修改
人工智能·opencv·计算机视觉
老艾的AI世界7 小时前
AI翻唱神器,一键用你喜欢的歌手翻唱他人的曲目(附下载链接)
人工智能·深度学习·神经网络·机器学习·ai·ai翻唱·ai唱歌·ai歌曲
DK221517 小时前
机器学习系列----关联分析
人工智能·机器学习
Robot2517 小时前
Figure 02迎重大升级!!人形机器人独角兽[Figure AI]商业化加速
人工智能·机器人·微信公众平台