找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
查看: 4996|回复: 0
打印 上一主题 下一主题
收起左侧

张正友相机标定源码 利用opencv实现

[复制链接]
跳转到指定楼层
楼主
张正友相机标定。利用opencv实现


单片机源程序如下:
  1. #include "opencv2/core/core.hpp"
  2. #include "opencv2/imgproc/imgproc.hpp"
  3. #include "opencv2/calib3d/calib3d.hpp"
  4. #include "opencv2/highgui/highgui.hpp"
  5. #include <iostream>
  6. #include <fstream>
  7. #include <iomanip>
  8. using namespace cv;
  9. using namespace std;

  10. void main()
  11. {
  12.         //ifstream fin("calibdata.txt"); /* 标定所用图像文件的路径 */
  13.         //ofstream fout("caliberation_result.txt");  /* 保存标定结果的文件 */
  14.         //                                                                                   //读取每一幅图像,从中提取出角点,然后对角点进行亚像素精确化       
  15.         //cout << "开始提取角点………………";
  16.         //int image_count = 0;  /* 图像数量 */
  17.         //Size image_size;  /* 图像的尺寸 */
  18.         //Size board_size = Size(4, 6);    /* 标定板上每行、列的角点数 */
  19.         //vector<Point2f> image_points_buf;  /* 缓存每幅图像上检测到的角点 */
  20.         //vector<vector<Point2f>> image_points_seq; /* 保存检测到的所有角点 */
  21.         //string filename;
  22.         //int count = -1;//用于存储角点个数。
  23.         //while (getline(fin, filename))
  24.         //{
  25.         //        image_count++;
  26.         //        // 用于观察检验输出
  27.         //        cout << "image_count = " << image_count << endl;
  28.         //        /* 输出检验*/
  29.         //        cout << "-->count = " << count;
  30.         //        Mat imageInput = imread(filename);

  31.         ofstream fout("caliberation_result.txt");  /* 保存标定结果的文件 */
  32.         cv::namedWindow("Image");
  33.         cv::Mat imageInput;
  34.         std::vector<std::string> filelist;//存放标定图片路径
  35.         int image_count = 0;  /* 图像数量 */
  36.         Size image_size;  /* 图像的尺寸 */
  37.         Size board_size = Size(4, 6);    /* 标定板上每行、列的角点数 */
  38.         vector<Point2f> image_points_buf;  /* 缓存每幅图像上检测到的角点 */
  39.         vector<vector<Point2f>> image_points_seq; /* 保存检测到的所有角点 */
  40.         int count = -1;//用于存储角点个数。
  41.                                                                           //生成路径,此处表示图片放在工程根目录下的chessboards文件夹
  42.         for (int i = 1; i <= 21; i++)
  43.         {
  44.                 std::stringstream str;//setw(int n)用来控制输出间隔
  45.                 str << "chessboards/chessboard" << std::setw(2) << std::setfill('0') << i << ".jpg";//图片的相对路径
  46.                 std::cout << str.str() << std::endl;

  47.                 filelist.push_back(str.str());
  48.                 imageInput = cv::imread(str.str());
  49.                 cv::imshow("Image", imageInput);


  50.                 //if (image_count == 1)  //读入第一张图片时获取图像宽高信息
  51.                 //{
  52.                 //        image_size.width = imageInput.cols;
  53.                 //        image_size.height = imageInput.rows;
  54.                 //        cout << "image_size.width = " << image_size.width << endl;
  55.                 //        cout << "image_size.height = " << image_size.height << endl;
  56.                 //}

  57.                 /* 提取角点 */
  58.                 //if (0 == findChessboardCorners(imageInput, board_size, image_points_buf))
  59.                 //{
  60.                 //        cout << "can not find chessboard corners!\n"; //找不到角点
  61.                 //        exit(1);
  62.                 //}
  63.                 //else
  64.                 //{
  65.                     findChessboardCorners(imageInput, board_size, image_points_buf);
  66.                         Mat view_gray;
  67.                         cvtColor(imageInput, view_gray, CV_RGB2GRAY);
  68.                         findChessboardCorners(imageInput, board_size, image_points_buf);
  69.                         /* 亚像素精确化 */
  70.                         find4QuadCornerSubpix(view_gray, image_points_buf, Size(5, 5)); //对粗提取的角点进行精确化
  71.                                                                                                                                                         //cornerSubPix(view_gray,image_points_buf,Size(5,5),Size(-1,-1),TermCriteria(CV_TERMCRIT_EPS+CV_TERMCRIT_ITER,30,0.1));
  72.                         image_points_seq.push_back(image_points_buf);  //保存亚像素角点
  73.                                                                                                                    /* 在图像上显示角点位置 */
  74.                         drawChessboardCorners(view_gray, board_size, image_points_buf, false); //用于在图片中标记角点
  75.                         imshow("Camera Calibration", view_gray);//显示图片
  76.                         waitKey(200);//暂停0.5S               
  77.                


  78.         }
  79.        
  80.         int total = image_points_seq.size();
  81.         cout << "total = " << total << endl;
  82.         int CornerNum = board_size.width*board_size.height;  //每张图片上总的角点数
  83.         for (int ii = 0; ii<total; ii++)
  84.         {
  85.                 if (0 == ii%CornerNum)// 24 是每幅图片的角点个数。此判断语句是为了输出 图片号,便于控制台观看
  86.                 {
  87.                         int i = -1;
  88.                         i = ii / CornerNum;
  89.                         int j = i + 1;
  90.                         cout << "--> 第 " << j << "图片的数据 --> : " << endl;
  91.                 }
  92.                 if (0 == ii % 3)        // 此判断语句,格式化输出,便于控制台查看
  93.                 {
  94.                         cout << endl;
  95.                 }
  96.                 else
  97.                 {
  98.                         cout.width(10);
  99.                 }
  100.                 //输出所有的角点
  101.                 cout << " -->" << image_points_seq[ii][0].x;
  102.                 cout << " -->" << image_points_seq[ii][0].y;
  103.         }
  104.         cout << "角点提取完成!\n";

  105.         //以下是摄像机标定
  106.         cout << "开始标定………………";
  107.         /*棋盘三维信息*/
  108.         Size square_size = Size(10, 10);  /* 实际测量得到的标定板上每个棋盘格的大小 */
  109.         vector<vector<Point3f>> object_points; /* 保存标定板上角点的三维坐标 */
  110.                                                                                    /*内外参数*/
  111.         Mat cameraMatrix = Mat(3, 3, CV_32FC1, Scalar::all(0)); /* 摄像机内参数矩阵 */
  112.         vector<int> point_counts;  // 每幅图像中角点的数量
  113.         Mat distCoeffs = Mat(1, 5, CV_32FC1, Scalar::all(0)); /* 摄像机的5个畸变系数:k1,k2,p1,p2,k3 */
  114.         vector<Mat> tvecsMat;  /* 每幅图像的旋转向量 */
  115.         vector<Mat> rvecsMat; /* 每幅图像的平移向量 */
  116.                                                   /* 初始化标定板上角点的三维坐标 */
  117.         int i, j, t;
  118.         for (t = 0; t<image_count; t++)
  119.         {
  120.                 vector<Point3f> tempPointSet;
  121.                 for (i = 0; i<board_size.height; i++)
  122.                 {
  123.                         for (j = 0; j<board_size.width; j++)
  124.                         {
  125.                                 Point3f realPoint;
  126.                                 /* 假设标定板放在世界坐标系中z=0的平面上 */
  127.                                 realPoint.x = i*square_size.width;
  128.                                 realPoint.y = j*square_size.height;
  129.                                 realPoint.z = 0;
  130.                                 tempPointSet.push_back(realPoint);
  131.                         }
  132.                 }
  133.                 object_points.push_back(tempPointSet);
  134.         }
  135.         /* 初始化每幅图像中的角点数量,假定每幅图像中都可以看到完整的标定板 */
  136.         for (i = 0; i<image_count; i++)
  137.         {
  138.                 point_counts.push_back(board_size.width*board_size.height);
  139.         }
  140.         /* 开始标定 */
  141.         calibrateCamera(object_points, image_points_seq, image_size, cameraMatrix, distCoeffs, rvecsMat, tvecsMat, 0);
  142.         cout << "标定完成!\n";
  143.         //对标定结果进行评价
  144.         cout << "开始评价标定结果………………\n";
  145.         double total_err = 0.0; /* 所有图像的平均误差的总和 */
  146.         double err = 0.0; /* 每幅图像的平均误差 */
  147.         vector<Point2f> image_points2; /* 保存重新计算得到的投影点 */
  148.         cout << "\t每幅图像的标定误差:\n";
  149.         fout << "每幅图像的标定误差:\n";
  150.         for (i = 0; i<image_count; i++)
  151.         {
  152.                 vector<Point3f> tempPointSet = object_points[i];
  153.                 /* 通过得到的摄像机内外参数,对空间的三维点进行重新投影计算,得到新的投影点 */
  154.                 projectPoints(tempPointSet, rvecsMat[i], tvecsMat[i], cameraMatrix, distCoeffs, image_points2);
  155.                 /* 计算新的投影点和旧的投影点之间的误差*/
  156.                 vector<Point2f> tempImagePoint = image_points_seq[i];
  157.                 Mat tempImagePointMat = Mat(1, tempImagePoint.size(), CV_32FC2);
  158.                 Mat image_points2Mat = Mat(1, image_points2.size(), CV_32FC2);
  159.                 for (int j = 0; j < tempImagePoint.size(); j++)
  160.                 {
  161.                         image_points2Mat.at<Vec2f>(0, j) = Vec2f(image_points2[j].x, image_points2[j].y);
  162.                         tempImagePointMat.at<Vec2f>(0, j) = Vec2f(tempImagePoint[j].x, tempImagePoint[j].y);
  163.                 }
  164.                 err = norm(image_points2Mat, tempImagePointMat, NORM_L2);
  165.                 total_err += err /= point_counts[i];
  166.                 std::cout << "第" << i + 1 << "幅图像的平均误差:" << err << "像素" << endl;
  167.                 fout << "第" << i + 1 << "幅图像的平均误差:" << err << "像素" << endl;
  168.         }
  169.         std::cout << "总体平均误差:" << total_err / image_count << "像素" << endl;
  170.         fout << "总体平均误差:" << total_err / image_count << "像素" << endl << endl;
  171.         std::cout << "评价完成!" << endl;
  172.         //保存定标结果         
  173.         std::cout << "开始保存定标结果………………" << endl;
  174.         Mat rotation_matrix = Mat(3, 3, CV_32FC1, Scalar::all(0)); /* 保存每幅图像的旋转矩阵 */
  175.         fout << "相机内参数矩阵:" << endl;
  176.         fout << cameraMatrix << endl << endl;
  177.         fout << "畸变系数:\n";
  178.         fout << distCoeffs << endl << endl << endl;
  179.         for (int i = 0; i<image_count; i++)
  180.         {
  181.                 fout << "第" << i + 1 << "幅图像的旋转向量:" << endl;
  182.                 fout << tvecsMat[i] << endl;
  183.                 /* 将旋转向量转换为相对应的旋转矩阵 */
  184.                 Rodrigues(tvecsMat[i], rotation_matrix);
  185.                 fout << "第" << i + 1 << "幅图像的旋转矩阵:" << endl;
  186.                 fout << rotation_matrix << endl;
  187.                 fout << "第" << i + 1 << "幅图像的平移向量:" << endl;
  188.                 fout << rvecsMat[i] << endl << endl;
  189.         }
  190.         std::cout << "完成保存" << endl;
  191.         fout << endl;
  192.         /************************************************************************
  193.         显示定标结果
  194.         *************************************************************************/
  195.         Mat mapx = Mat(image_size, CV_32FC1);
  196.         Mat mapy = Mat(image_size, CV_32FC1);
  197.         Mat R = Mat::eye(3, 3, CV_32F);
  198.         std::cout << "保存矫正图像" << endl;
  199.         string imageFileName;
  200.         std::stringstream StrStm;
  201.         for (int i = 0; i != image_count; i++)
  202.         {
  203.                 std::cout << "Frame #" << i + 1 << "..." << endl;
  204.                 initUndistortRectifyMap(cameraMatrix, distCoeffs, R, cameraMatrix, image_size, CV_32FC1, mapx, mapy);
  205.                 StrStm.clear();
  206.                 imageFileName.clear();
  207.                 string filePath = "chess";
  208.                 StrStm << i + 1;
  209.                 StrStm >> imageFileName;
  210.                 filePath += imageFileName;
  211.                 filePath += ".bmp";
  212.                 Mat imageSource = imread(filePath);
  213.                 Mat newimage = imageSource.clone();
  214.                 //另一种不需要转换矩阵的方式
  215.                 //undistort(imageSource,newimage,cameraMatrix,distCoeffs);
  216.                 remap(imageSource, newimage, mapx, mapy, INTER_LINEAR);
  217.                 StrStm.clear();
  218.                 filePath.clear();
  219.                 StrStm << i + 1;
  220.                 StrStm >> imageFileName;
  221.                 imageFileName += "_d.jpg";
  222.                 imwrite(imageFileName, newimage);
  223.         }
  224.         std::cout << "保存结束" << endl;
  225.         return;
  226. }
复制代码

所有资料51hei提供下载:
张正友相机标定.zip (6.57 MB, 下载次数: 25)


评分

参与人数 1黑币 +50 收起 理由
admin + 50 共享资料的黑币奖励!

查看全部评分

分享到:  QQ好友和群QQ好友和群 QQ空间QQ空间 腾讯微博腾讯微博 腾讯朋友腾讯朋友
收藏收藏1 分享淘帖 顶 踩
回复

使用道具 举报

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

本版积分规则

手机版|小黑屋|51黑电子论坛 |51黑电子论坛6群 QQ 管理员QQ:125739409;技术交流QQ群281945664

Powered by 单片机教程网

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