opencv的一个神经网络例程(转自http://blog.sina.com.cn/s/blog_61e10f020101bv94.html)

opencv的一个神经网络例程（转）

 (2013-07-19 00:45:34)


转载▼

分类： OPENCV学习笔记

//The example of using BPNetwork in OpenCV

//Coded by L. Wei

#include

#include

#include

#include

#include

using namespace std;

using namespace cv;

int main()

{

 //Setup the BPNetwork

 CvANN_MLP bp;

 // Set up BPNetwork's parameters

 CvANN_MLP_TrainParams params;

 params.train_method=CvANN_MLP_TrainParams::BACKPROP;

 params.bp_dw_scale=0.1;

 params.bp_moment_scale=0.1;

 //params.train_method=CvANN_MLP_TrainParams::RPROP;

 //params.rp_dw0 = 0.1;

 //params.rp_dw_plus = 1.2;

 //params.rp_dw_minus = 0.5;

 //params.rp_dw_min = FLT_EPSILON;

 //params.rp_dw_max = 50.;

 // Set up training data

 float labels[3][5] = {{0,0,0,0,0},{1,1,1,1,1},{0,0,0,0,0}};

 Mat labelsMat(3, 5, CV_32FC1, labels);

 float trainingData[3][5] = { {1,2,3,4,5},{111,112,113,114,115}, {21,22,23,24,25} };

 Mat trainingDataMat(3, 5, CV_32FC1, trainingData);

 Mat layerSizes=(Mat_(1,5) << 5,2,2,2,5);

 bp.create(layerSizes,CvANN_MLP::SIGMOID_SYM);//CvANN_MLP::SIGMOID_SYM

                                            //CvANN_MLP::GAUSSIAN

                                            //CvANN_MLP::IDENTITY

 bp.train(trainingDataMat, labelsMat, Mat(),Mat(), params);

 // Data for visual representation

 int width = 512, height = 512;

 Mat image = Mat::zeros(height, width, CV_8UC3);

 Vec3b green(0,255,0), blue (255,0,0);

 // Show the decision regions given by the SVM

 for (int i = 0; i < image.rows; ++i)

  for (int j = 0; j < image.cols; ++j)

  {

   Mat sampleMat = (Mat_(1,5) << i,j,0,0,0);

   Mat responseMat;

   bp.predict(sampleMat,responseMat);

   float* p=responseMat.ptr(0);

   int response=0;

   for(int i=0;i<5;i++){

   // cout<<p[i]<<" ";

    response+=p[i];

   }

   if (response >2)

    image.at(j, i)  = green;

   else 

    image.at(j, i)  = blue;

  }

  // Show the training data

  int thickness = -1;

  int lineType = 8;

  circle( image, Point(501,  10), 5, Scalar(  0,   0,   0), thickness, lineType);

  circle( image, Point(255,  10), 5, Scalar(255, 255, 255), thickness, lineType);

  circle( image, Point(501, 255), 5, Scalar(255, 255, 255), thickness, lineType);

  circle( image, Point( 10, 501), 5, Scalar(255, 255, 255), thickness, lineType);

  imwrite("result.png", image);        // save the image

  imshow("BP Simple Example", image); // show it to the user

  waitKey(0);

}