码迷，mamicode.com
首页 > 其他好文 > 详细

背景建模技术（七）：预处理（PreProcessor）模块

时间：2015-05-18 14:38:43      阅读：171      评论：0      收藏：0      [点我收藏+]

标签：

预处理（PreProcessor）模块是BgsLibrary中一个必选的模块，是真正进入背景建模算法的“预处理”过程，其主要功能包括‘去模糊’、‘获得灰度图’、‘应用Canny算子‘等可选模块。

 

下面给出源码：

 

 

[cpp] view plaincopyprint?
  1. #include "PreProcessor.h"  
  2.   
  3. namespace bgslibrary  
  4. {  
  5.   PreProcessor::PreProcessor() : firstTime(true), equalizeHist(false), gaussianBlur(false)  
  6.   {  
  7.     std::cout << "PreProcessor()" << std::endl;  
  8.   }  
  9.   
  10.   PreProcessor::~PreProcessor()  
  11.   {  
  12.     std::cout << "~PreProcessor()" << std::endl;  
  13.   }  
  14.   
  15.   void PreProcessor::setEqualizeHist(bool value)  
  16.   {  
  17.     equalizeHist = value;  
  18.   }  
  19.   
  20.   void PreProcessor::setGaussianBlur(bool value)  
  21.   {  
  22.     gaussianBlur = value;  
  23.   }  
  24.   
  25.   cv::Mat PreProcessor::getGrayScale()  
  26.   {  
  27.     return img_gray.clone();  
  28.   }  
  29.   
  30.   void PreProcessor::process(const cv::Mat &img_input, cv::Mat &img_output)  
  31.   {  
  32.     if (img_input.empty())  
  33.       return;  
  34.   
  35.     loadConfig();  
  36.   
  37.     if (firstTime)  
  38.       saveConfig();  
  39.   
  40.     img_input.copyTo(img_output);  
  41.   
  42.     // Converts image from one color space to another  
  43.     // http://opencv.willowgarage.com/documentation/cpp/miscellaneous_image_transformations.html#cv-cvtcolor  
  44.     cv::cvtColor(img_input, img_gray, CV_BGR2GRAY);  
  45.     //img_gray.copyTo(img_output);  
  46.   
  47.     // Equalizes the histogram of a grayscale image  
  48.     // http://opencv.willowgarage.com/documentation/cpp/histograms.html#cv-equalizehist  
  49.     if (equalizeHist)  
  50.       cv::equalizeHist(img_output, img_output);  
  51.   
  52.     // Smoothes image using a Gaussian filter  
  53.     // http://opencv.willowgarage.com/documentation/cpp/imgproc_image_filtering.html#GaussianBlur  
  54.     if (gaussianBlur)  
  55.       cv::GaussianBlur(img_output, img_output, cv::Size(7, 7), 1.5);  
  56.   
  57.     if (enableShow)  
  58.       cv::imshow("Pre Processor", img_output);  
  59.   
  60.     firstTime = false;  
  61.   }  
  62.   
  63.   void PreProcessor::rotate(const cv::Mat &img_input, cv::Mat &img_output, float angle)  
  64.   {  
  65.     IplImage* image = new IplImage(img_input);  
  66.   
  67.     //IplImage *rotatedImage = cvCreateImage(cvSize(480,320), IPL_DEPTH_8U, image->nChannels);  
  68.     //IplImage *rotatedImage = cvCreateImage(cvSize(image->width,image->height), IPL_DEPTH_8U, image->nChannels);  
  69.     IplImage* rotatedImage = cvCreateImage(cvSize(image->height, image->width), IPL_DEPTH_8U, image->nChannels);  
  70.   
  71.     CvPoint2D32f center;  
  72.     //center.x = 160;  
  73.     //center.y = 160;  
  74.     center.x = (image->height / 2);  
  75.     center.y = (image->width / 2);  
  76.   
  77.     CvMat* mapMatrix = cvCreateMat(2, 3, CV_32FC1);  
  78.   
  79.     cv2DRotationMatrix(center, angle, 1.0, mapMatrix);  
  80.     cvWarpAffine(image, rotatedImage, mapMatrix, CV_INTER_LINEAR + CV_WARP_FILL_OUTLIERS, cvScalarAll(0));  
  81.   
  82.     cv::Mat img_rot(rotatedImage);  
  83.     img_rot.copyTo(img_output);  
  84.   
  85.     cvReleaseImage(&image);  
  86.     cvReleaseImage(&rotatedImage);  
  87.     cvReleaseMat(&mapMatrix);  
  88.   }  
  89.   
  90.   void PreProcessor::applyCanny(const cv::Mat &img_input, cv::Mat &img_output)  
  91.   {  
  92.     if (img_input.empty())  
  93.       return;  
  94.   
  95.     //------------------------------------------------------------------  
  96.     // Canny  
  97.     // Finds edges in an image using Canny algorithm.  
  98.     // http://opencv.willowgarage.com/documentation/cpp/imgproc_feature_detection.html#cv-canny  
  99.     //------------------------------------------------------------------  
  100.   
  101.     cv::Mat img_canny;  
  102.     cv::Canny(  
  103.       img_input, // image ?Single-channel 8-bit input image  
  104.       img_canny,  // edges ?The output edge map. It will have the same size and the same type as image  
  105.       100,       // threshold1 ?The first threshold for the hysteresis procedure  
  106.       200);      // threshold2 ?The second threshold for the hysteresis procedure  
  107.     cv::threshold(img_canny, img_canny, 128, 255, cv::THRESH_BINARY_INV);  
  108.   
  109.     img_canny.copyTo(img_output);  
  110.   }  
  111.   
  112.   void PreProcessor::saveConfig()  
  113.   {  
  114.     CvFileStorage* fs = cvOpenFileStorage("./config/PreProcessor.xml", 0, CV_STORAGE_WRITE);  
  115.   
  116.     cvWriteInt(fs, "equalizeHist", equalizeHist);  
  117.     cvWriteInt(fs, "gaussianBlur", gaussianBlur);  
  118.     cvWriteInt(fs, "enableShow", enableShow);  
  119.   
  120.     cvReleaseFileStorage(&fs);  
  121.   }  
  122.   
  123.   void PreProcessor::loadConfig()  
  124.   {  
  125.     CvFileStorage* fs = cvOpenFileStorage("./config/PreProcessor.xml", 0, CV_STORAGE_READ);  
  126.   
  127.     equalizeHist = cvReadIntByName(fs, 0, "equalizeHist", false);  
  128.     gaussianBlur = cvReadIntByName(fs, 0, "gaussianBlur", false);  
  129.     enableShow = cvReadIntByName(fs, 0, "enableShow", true);  
  130.   
  131.     cvReleaseFileStorage(&fs);  
  132.   }  
  133. }  


最后给出此模块的流程框架图供大家参考：

 

 

技术分享

 

背景建模技术（七）：预处理（PreProcessor）模块

标签：

原文地址：http://www.cnblogs.com/ywsoftware/p/4511787.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！