前景物体提取

  金牌会员 | 2024-12-22 02:44:15 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 807|帖子 807|积分 2421

参考:精选课:C++完备的实现双目摄像头图像采集、双目摄像头畸变改正、前景物体提取、生成视差图、深度图、PCL点云图
前景物体提取是盘算机视觉中的一个重要技术,可以用于视频监控、虚拟现实和盘算机视觉等范畴。
1.前景物体提取的原理

前景物体提取是将摄像机拍摄到的图像中的前景物体(如移动的人、车辆等)从配景中分离出来的过程。其原理是利用帧间差分和配景建模两个步骤。帧间差分是通过相邻帧之间像素点灰度值的差别来检测出活动目标,配景建模是通过不断更新配景图像来适应场景的变化。
2.实现步骤

步骤1:读取摄像机视频并初始化,使用OpenCV库来读取摄像机视频:
  1. cv::VideoCapture cap(0);
  2. if (!cap.isOpened()) {
  3.    std::cout << "Cannot open camera" << std::endl;
  4.    return -1;
  5. }
  6. cv::Mat frame;
  7. cap.read(frame);
复制代码
步骤2:配景建模,界说一个配景图像和一个学习率,初始值为0.01。在每一帧中,将当前帧与配景图像相减,得到一个差分图像。
  1. cv::Mat background;
  2. double learning_rate = 0.01;
  3. // 第一帧作为背景图像
  4. background = frame.clone();
  5. // 对于后面的帧,逐像素地计算背景图像
  6. while (true) {
  7.     cap.read(frame);
  8.     // 将当前帧与背景图像相减,得到一个差分图像
  9.     cv::Mat diff;
  10.     cv::absdiff(frame, background, diff);
  11.     // 根据差分图像更新背景图像
  12.     for (int i = 0; i < diff.rows; i++) {
  13.         for (int j = 0; j < diff.cols; j++) {
  14.             cv::Vec3b pixel = diff.at<cv::Vec3b>(i, j);
  15.             if (pixel[0] > 50 || pixel[1] > 50 || pixel[2] > 50) {
  16.                 // 更新背景像素
  17.                 cv::Vec3b background_pixel = background.at<cv::Vec3b>(i, j);
  18.                 cv::Vec3b frame_pixel = frame.at<cv::Vec3b>(i, j);
  19.                 background_pixel[0] = (1 - learning_rate) * background_pixel[0] + learning_rate * frame_pixel[0];
  20.                 background_pixel[1] = (1 - learning_rate) * background_pixel[1] + learning_rate * frame_pixel[1];
  21.                 background_pixel[2] = (1 - learning_rate) * background_pixel[2] + learning_rate * frame_pixel[2];
  22.                 background.at<cv::Vec3b>(i, j) = background_pixel;
  23.             }
  24.         }
  25.     }
  26. }
复制代码
步骤3:帧间差分,将当前帧与配景图像相减,得到差分图像。然后将差分图像进行二值化处理,得到前景物体掩模。
  1. // 帧间差分
  2. cv::Mat diff;
  3. cv::absdiff(frame, background, diff);
  4. // 二值化处理,得到前景掩模
  5. cv::Mat foreground_mask;
  6. cv::threshold(diff, foreground_mask, 50, 255, cv::THRESH_BINARY);
复制代码
步骤4:过滤掉小的前景物体,使用形态学操作对前景掩模进行处理,去除噪声和小物体。
  1. // 使用开运算去除噪声和小物体
  2. cv::Mat kernel = cv::getStructuringElement(cv::MORPH_ELLIPSE, cv::Size(5, 5));
  3. cv::Mat foreground_mask_filtered;
  4. cv::morphologyEx(foreground_mask, foreground_mask_filtered, cv::MORPH_OPEN, kernel);
复制代码
步骤5:表现结果,将原始图像和前景掩模相乘,得到只有前景物体的图像。
  1. // 显示结果
  2. cv::Mat result = frame.clone();
  3. cv::Mat foreground_image = cv::Mat::zeros(frame.size(), frame.type());
  4. frame.copyTo(foreground_image, foreground_mask_filtered);
  5. cv::imshow("Original Image", frame);
  6. cv::imshow("Foreground Mask", foreground_mask_filtered);
  7. cv::imshow("Foreground Image", foreground_image);
  8. cv::waitKey(30);
复制代码
3.完备代码

演示了对单个摄像头的前景物体提取,双目摄像机须要对两个摄像头的同一幅图片场景分别做前景物体提取,然后作为左右视图对其进行进一步畸变改正。
  1. #include <iostream>
  2. #include <opencv2/opencv.hpp>
  3. int main() {
  4.     // 打开摄像机
  5.     cv::VideoCapture cap(0);
  6.     if (!cap.isOpened()) {
  7.         std::cout << "Cannot open camera" << std::endl;
  8.         return -1;
  9.     }
  10.     // 初始化
  11.     cv::Mat frame;
  12.     cap.read(frame);
  13.     // 背景建模
  14.     cv::Mat background;
  15.     double learning_rate = 0.01;
  16.     background = frame.clone();
  17.     while (true) {
  18.         // 读取帧
  19.         cap.read(frame);
  20.         // 背景建模
  21.         cv::Mat diff;
  22.         cv::absdiff(frame, background, diff);
  23.         for (int i = 0; i < diff.rows; i++) {
  24.             for (int j = 0; j < diff.cols; j++) {
  25.                 cv::Vec3b pixel = diff.at<cv::Vec3b>(i, j);
  26.                 if (pixel[0] > 50 || pixel[1] > 50 || pixel[2] > 50) {
  27.                     cv::Vec3b background_pixel = background.at<cv::Vec3b>(i, j);
  28.                     cv::Vec3b frame_pixel = frame.at<cv::Vec3b>(i, j);
  29.                     background_pixel[0] = (1 - learning_rate) * background_pixel[0] + learning_rate * frame_pixel[0];
  30.                     background_pixel[1] = (1 - learning_rate) * background_pixel[1] + learning_rate * frame_pixel[1];
  31.                     background_pixel[2] = (1 - learning_rate) * background_pixel[2] + learning_rate * frame_pixel[2];
  32.                     background.at<cv::Vec3b>(i, j) = background_pixel;
  33.                 }
  34.             }
  35.         }
  36.         // 帧间差分
  37.         cv::Mat diff2;
  38.         cv::absdiff(frame, background, diff2);
  39.         cv::Mat foreground_mask;
  40.         cv::threshold(diff2, foreground_mask, 50, 255, cv::THRESH_BINARY);
  41.         // 去除噪声和小物体
  42.         cv::Mat kernel = cv::getStructuringElement(cv::MORPH_ELLIPSE, cv::Size(5, 5));
  43.         cv::Mat foreground_mask_filtered;
  44.         cv::morphologyEx(foreground_mask, foreground_mask_filtered, cv::MORPH_OPEN, kernel);
  45.         // 显示结果
  46.         cv::Mat result = frame.clone();
  47.         cv::Mat foreground_image = cv::Mat::zeros(frame.size(), frame.type());
  48.         frame.copyTo(foreground_image, foreground_mask_filtered);
  49.         cv::imshow("Original Image", frame);
  50.         cv::imshow("Foreground Mask", foreground_mask_filtered);
  51.         cv::imshow("Foreground Image", foreground_image);
  52.         cv::waitKey(30);
  53.     }
  54.     // 释放资源
  55.     cap.release();
  56.     cv::destroyAllWindows();
  57.     return 0;
  58. }
复制代码
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表