1. 程式人生 > >opencv對影象進行標定

opencv對影象進行標定

簡介

  本篇是使用opencv函式:cvFindChessboardCorners、cvFindCornerSubPix、cvDrawChessboardCorners,來找到、優化並顯示出來標定棋盤
圖片的角點。
  關於這三個函式得講解看,可以參考:http://www.360doc.cn/article/10724725_367761079.html

角點檢測

具體程式碼

  具體程式碼如下:
#include <opencv2/opencv.hpp>
#include <stdio.h>
using namespace cv;
using namespace std;
 
int boardWidth, boardHeight;
int boardTotal;
CvSize boardSize;
CvPoint2D32f * image_points_buf;
Mat srcColor, srcGray;
IplImage srcIp;
CvMat cvmatSrc;
int nowNumber, found;
char* picName;
 
void initFindCorner(){
    boardSize = cvSize(10, 7);
    boardWidth = boardSize.width;   
    boardHeight = boardSize.height;
    boardTotal = boardWidth*boardHeight;
 
    image_points_buf = new CvPoint2D32f[boardTotal];
 
    srcColor = imread(picName);
    imshow("源影象", srcColor);
 
    cvtColor(srcColor, srcGray, COLOR_BGR2GRAY);
    imshow("灰階圖", srcGray);
    srcIp = srcGray;
    cvmatSrc = srcColor;
}
 
void findCornersWork(){
    found=cvFindChessboardCorners(&srcIp, boardSize, image_points_buf, \
                 &nowNumber, CV_CALIB_CB_ADAPTIVE_THRESH|CV_CALIB_CB_FILTER_QUADS);
    printf("捕獲角點數量:%d\n", nowNumber);
 
    cvFindCornerSubPix(&srcIp, image_points_buf, nowNumber, cvSize(11,11), cvSize(-1,-1),
                                cvTermCriteria(CV_TERMCRIT_EPS+CV_TERMCRIT_ITER,30,0.1));
    cvDrawChessboardCorners(&cvmatSrc, boardSize, image_points_buf, nowNumber, found);
    imshow("角點標識圖", srcColor);
}
 
int main(int argc, char* argv[]){
 
    if(argc < 2){
        cout << "Please input Picture name !!\n" << endl;
        return -1;
    }
    picName = argv[1];
    initFindCorner();
    findCornersWork();
 
    waitKey();
 
    return 0;
}
程式碼講解
  1、初始化
void initFindCorner(){
    boardSize = cvSize(10, 7);
    boardWidth = boardSize.width;   
    boardHeight = boardSize.height;
    boardTotal = boardWidth*boardHeight;
 
    image_points_buf = new CvPoint2D32f[boardTotal];
 
    srcColor = imread(picName);
    imshow("源影象", srcColor);
 
    cvtColor(srcColor, srcGray, COLOR_BGR2GRAY);
    imshow("灰階圖", srcGray);
    srcIp = srcGray;
    cvmatSrc = srcColor;
}
首先設定預先設定圖片的角點個數,本例使用的棋盤圖片角點個數為:10X7,建立儲存角點的結構:image_points_buf,接著匯入棋盤圖片,
並轉為灰階影象。
  2、角點檢測和顯示
void findCornersWork(){
    found=cvFindChessboardCorners(&srcIp, boardSize, image_points_buf, \
                 &nowNumber, CV_CALIB_CB_ADAPTIVE_THRESH|CV_CALIB_CB_FILTER_QUADS);
    printf("捕獲角點數量:%d\n", nowNumber);
 
    cvFindCornerSubPix(&srcIp, image_points_buf, nowNumber, cvSize(11,11), cvSize(-1,-1),
                                cvTermCriteria(CV_TERMCRIT_EPS+CV_TERMCRIT_ITER,30,0.1));
    cvDrawChessboardCorners(&cvmatSrc, boardSize, image_points_buf, nowNumber, found);
    imshow("角點標識圖", srcColor);
}
    使用cvFindChessboardCorners進行角點檢測,檢測到的角點儲存在image_points_buf,檢測到的角點數量儲存在nowNumber,如果nowNumber的值,和
實際圖片上的角點數量相等,就表示角點檢測成功。
  接著cvFindCornerSubPix、cvDrawChessboardCorners將這些檢測到的角點位置在圖片:cvmatSrc上,顯示輸出。

結果顯示

  顯示的結果如下:
        

畸變校正

  前面已經講解了如果找到棋盤標點圖片的角點,這裡在此基礎上,繼續進行後續的校正處理。
首先是找到多張圖片的角點,接著將這些角點匯入到函式cvCalibrateCamera2,進行camera內參數矩陣和畸變係數向量的生成。通過cvInitUndistortMap,
利用之前生成的內參數矩陣和畸變向量,計算出畸變對映到mapx和mapy中。最後cvRemap利用mapx、mapy對輸入影象進行畸變校正。
  可以參考文件:http://blog.csdn.net/guvcolie/article/details/7454632

具體程式碼

#include <opencv2/opencv.hpp>
#include <stdio.h>
using namespace cv;
using namespace std;
 
int main(int argc, char* argv[]){
    int cube_length=10;
    int cam_Dx = 100; //橫軸方向長度
    int cam_Dy = 100; //縱軸方向長度
    int number_image = 7;
    int a=1;
    int number_image_copy= 7;
    CvSize board_size=cvSize(10,7);
    int board_width=board_size.width;
    int board_height=board_size.height;
    int total_per_image=board_width*board_height;
    CvPoint2D32f * image_points_buf = new CvPoint2D32f[total_per_image];
    CvMat * image_points=cvCreateMat(number_image*total_per_image,2,CV_32FC1);//影象座標系
    CvMat * object_points=cvCreateMat(number_image*total_per_image,3,CV_32FC1);//世界座標系
    CvMat * point_counts=cvCreateMat(number_image,1,CV_32SC1); //角點存放位置
    CvMat * intrinsic_matrix=cvCreateMat(3,3,CV_32FC1);  //內參數矩陣
    CvMat * distortion_coeffs=cvCreateMat(4,1,CV_32FC1); //畸變係數向量
    char picName[7][10] = {"1.jpg", "2.jpg", "3.jpg", "4.jpg", "5.jpg", "6.jpg", "7.jpg"};
    IplImage * show;
 
    int count;
    int found;
    int step;
    int successes=0;
 
 
    while(a<=number_image_copy){
        show=cvLoadImage(picName[a-1],-1);
     
        found=cvFindChessboardCorners(show,board_size,image_points_buf,&count,
                CV_CALIB_CB_ADAPTIVE_THRESH|CV_CALIB_CB_FILTER_QUADS);
        if(found==0){      
            cout<<"第"<<a<<"幀圖片無法找到棋盤格所有角點!\n\n";
            cvNamedWindow("RePlay",1);
            cvShowImage("RePlay",show);
            cvWaitKey(0);
 
        }else{
            cout<<"第"<<a<<"幀影象成功獲得"<<count<<"個角點...\n";
 
            IplImage * gray_image= cvCreateImage(cvGetSize(show),8,1);
 
            cvCvtColor(show,gray_image,CV_BGR2GRAY);
 
            cout<<"獲取源影象灰度圖過程完成...\n";
            cvFindCornerSubPix(gray_image,image_points_buf,count,cvSize(11,11),cvSize(-1,-1),
                cvTermCriteria(CV_TERMCRIT_EPS+CV_TERMCRIT_ITER,30,0.1));
            cout<<"灰度圖亞畫素化過程完成...\n";
            cvDrawChessboardCorners(show,board_size,image_points_buf,count,found);
            cout<<"在源影象上繪製角點過程完成...\n\n";
        }
 
        if(total_per_image==count){
            step=successes*total_per_image;
            for(int i=step,j=0;j<total_per_image;++i,++j){
                CV_MAT_ELEM(*image_points,float,i,0)=image_points_buf[j].x; 
 
                CV_MAT_ELEM(*image_points,float,i,1)=image_points_buf[j].y;// 求完每個角點橫縱座標值都存在image_point_buf裡
                CV_MAT_ELEM(*object_points,float,i,0)=(float)((j/cube_length) * cam_Dx);
                CV_MAT_ELEM(*object_points,float,i,1)=(float)((j%cube_length) * cam_Dy);
                CV_MAT_ELEM(*object_points,float,i,2)=0.0f;
            }
            CV_MAT_ELEM(*point_counts,int,successes,0)=total_per_image;
            successes++;
        }
        a++;
    }
    cout<<"*********************************************\n";
    cout<<number_image<<"幀圖片中,標定成功的圖片為"<<successes<<"幀...\n";
    cout<<number_image<<"幀圖片中,標定失敗的圖片為"<<number_image-successes<<"幀...\n\n";
    cout<<"*********************************************\n\n";
 
    IplImage * show_colie;
    show_colie = show;
 
 
    CvMat * object_points2=cvCreateMat(successes*total_per_image,3,CV_32FC1);
 
    CvMat * image_points2=cvCreateMat(successes*total_per_image,2,CV_32FC1);
    CvMat * point_counts2=cvCreateMat(successes,1,CV_32SC1);
    for(int i=0;i<successes*total_per_image;++i){
        CV_MAT_ELEM(*image_points2,float,i,0)=CV_MAT_ELEM(*image_points,float,i,0);//用來儲存角點提取成功的影象的角點
        CV_MAT_ELEM(*image_points2,float,i,1)=CV_MAT_ELEM(*image_points,float,i,1);
        CV_MAT_ELEM(*object_points2,float,i,0)=CV_MAT_ELEM(*object_points,float,i,0);
        CV_MAT_ELEM(*object_points2,float,i,1)=CV_MAT_ELEM(*object_points,float,i,1);
        CV_MAT_ELEM(*object_points2,float,i,2)=CV_MAT_ELEM(*object_points,float,i,2);
    }
 
    for(int i=0;i<successes;++i){
        CV_MAT_ELEM(*point_counts2,int,i,0)=CV_MAT_ELEM(*point_counts,int,i,0);
    }
 
 
    cvReleaseMat(&object_points);
    cvReleaseMat(&image_points);
    cvReleaseMat(&point_counts);
 
    CV_MAT_ELEM(*intrinsic_matrix,float,0,0)=1.0f;
    CV_MAT_ELEM(*intrinsic_matrix,float,1,1)=1.0f;
 
    cvCalibrateCamera2(object_points2,image_points2,point_counts2,cvGetSize(show_colie),
            intrinsic_matrix,distortion_coeffs,NULL,NULL,0);
 
    cvSave("Intrinsics.xml",intrinsic_matrix);
    cvSave("Distortion.xml",distortion_coeffs);
 
    cout<<"攝像機矩陣、畸變係數向量已經分別儲存在名為Intrinsics.xml、Distortion.xml文件中\n\n";
 
    CvMat * intrinsic=(CvMat *)cvLoad("Intrinsics.xml");
    CvMat * distortion=(CvMat *)cvLoad("Distortion.xml");
 
    IplImage * mapx=cvCreateImage(cvGetSize(show_colie),IPL_DEPTH_32F,1);
    IplImage * mapy=cvCreateImage(cvGetSize(show_colie),IPL_DEPTH_32F,1);
 
    cvInitUndistortMap(intrinsic,distortion,mapx,mapy);
 
    cvNamedWindow("原始影象",1);
    cvNamedWindow("非畸變影象",1);
 
    show_colie = cvLoadImage(argv[1],-1);
    IplImage * clone=cvCloneImage(show_colie);
    cvShowImage("原始影象",show_colie);
    cvRemap(clone,show_colie,mapx,mapy);
    cvReleaseImage(&clone);
    cvShowImage("非畸變影象",show_colie);
 
    cvWaitKey(0);
 
    return 0;
}

程式碼講解

  1、首先這裡是使用了了7張棋盤圖片用來標定,所以cvFindChessboardCorners函式,會依次尋找7次角點。如果找到角點成功,則將對應結果儲存到
image_points、object_points中,注意儲存到object_points的時候需要做計算:(float)((j/cube_length) * cam_Dx);
while(a<=number_image_copy){
    show=cvLoadImage(picName[a-1],-1);
 
    found=cvFindChessboardCorners(show,board_size,image_points_buf,&count,
            CV_CALIB_CB_ADAPTIVE_THRESH|CV_CALIB_CB_FILTER_QUADS);
    if(found==0){      
        cout<<"第"<<a<<"幀圖片無法找到棋盤格所有角點!\n\n";
        cvNamedWindow("RePlay",1);
        cvShowImage("RePlay",show);
        cvWaitKey(0);
 
    }else{
        cout<<"第"<<a<<"幀影象成功獲得"<<count<<"個角點...\n";
 
        IplImage * gray_image= cvCreateImage(cvGetSize(show),8,1);
 
        cvCvtColor(show,gray_image,CV_BGR2GRAY);
 
        cout<<"獲取源影象灰度圖過程完成...\n";
        cvFindCornerSubPix(gray_image,image_points_buf,count,cvSize(11,11),cvSize(-1,-1),
            cvTermCriteria(CV_TERMCRIT_EPS+CV_TERMCRIT_ITER,30,0.1));
        cout<<"灰度圖亞畫素化過程完成...\n";
        cvDrawChessboardCorners(show,board_size,image_points_buf,count,found);
        cout<<"在源影象上繪製角點過程完成...\n\n";
    }
 
    if(total_per_image==count){
        step=successes*total_per_image;
        for(int i=step,j=0;j<total_per_image;++i,++j){
            CV_MAT_ELEM(*image_points,float,i,0)=image_points_buf[j].x; 
 
            CV_MAT_ELEM(*image_points,float,i,1)=image_points_buf[j].y;// 求完每個角點橫縱座標值都存在image_point_buf裡
            CV_MAT_ELEM(*object_points,float,i,0)=(float)((j/cube_length) * cam_Dx);
            CV_MAT_ELEM(*object_points,float,i,1)=(float)((j%cube_length) * cam_Dy);
            CV_MAT_ELEM(*object_points,float,i,2)=0.0f;
        }
        CV_MAT_ELEM(*point_counts,int,successes,0)=total_per_image;
        successes++;
    }
    a++;
}
    其中successes用來儲存,需找角點成功的次數,如果7次都成功,則successes為7。
  2、將找到的角點資訊,重新儲存到image_points2和object_points2中,利用cvCalibrateCamera2來計算矩陣、向量係數到intrinsic_matrix、distortion_coeffs,本儲存到本地檔案。
IplImage * show_colie;
show_colie = show;
 
 
CvMat * object_points2=cvCreateMat(successes*total_per_image,3,CV_32FC1);
 
CvMat * image_points2=cvCreateMat(successes*total_per_image,2,CV_32FC1);
CvMat * point_counts2=cvCreateMat(successes,1,CV_32SC1);
for(int i=0;i<successes*total_per_image;++i){
    CV_MAT_ELEM(*image_points2,float,i,0)=CV_MAT_ELEM(*image_points,float,i,0);//用來儲存角點提取成功的影象的角點
    CV_MAT_ELEM(*image_points2,float,i,1)=CV_MAT_ELEM(*image_points,float,i,1);
    CV_MAT_ELEM(*object_points2,float,i,0)=CV_MAT_ELEM(*object_points,float,i,0);
    CV_MAT_ELEM(*object_points2,float,i,1)=CV_MAT_ELEM(*object_points,float,i,1);
    CV_MAT_ELEM(*object_points2,float,i,2)=CV_MAT_ELEM(*object_points,float,i,2);
}
 
for(int i=0;i<successes;++i){
    CV_MAT_ELEM(*point_counts2,int,i,0)=CV_MAT_ELEM(*point_counts,int,i,0);
}
 
 
cvReleaseMat(&object_points);
cvReleaseMat(&image_points);
cvReleaseMat(&point_counts);
 
CV_MAT_ELEM(*intrinsic_matrix,float,0,0)=1.0f;
CV_MAT_ELEM(*intrinsic_matrix,float,1,1)=1.0f;
 
cvCalibrateCamera2(object_points2,image_points2,point_counts2,cvGetSize(show_colie),
        intrinsic_matrix,distortion_coeffs,NULL,NULL,0);
 
cvSave("Intrinsics.xml",intrinsic_matrix);
cvSave("Distortion.xml",distortion_coeffs);
  3、函式cvInitUndistortMap和cvRemap,通過之前計算的矩陣、向量係數,對需要校正的影象:show_colie進行處理,並分別顯示出來。
IplImage * mapx=cvCreateImage(cvGetSize(show_colie),IPL_DEPTH_32F,1);
IplImage * mapy=cvCreateImage(cvGetSize(show_colie),IPL_DEPTH_32F,1);
 
cvInitUndistortMap(intrinsic,distortion,mapx,mapy);
 
cvNamedWindow("原始影象",1);
cvNamedWindow("非畸變影象",1);
 
show_colie = cvLoadImage(argv[1],-1);
IplImage * clone=cvCloneImage(show_colie);
cvShowImage("原始影象",show_colie);
cvRemap(clone,show_colie,mapx,mapy);
cvReleaseImage(&clone);
cvShowImage("非畸變影象",show_colie);
video畸變校正
  在之前的基礎上,修改被校正的輸入即可,簡單的話,在前一個例子中,將如下程式碼:
    IplImage * mapx=cvCreateImage(cvGetSize(show_colie),IPL_DEPTH_32F,1);
    IplImage * mapy=cvCreateImage(cvGetSize(show_colie),IPL_DEPTH_32F,1);
 
    cvInitUndistortMap(intrinsic,distortion,mapx,mapy);
 
    cvNamedWindow("原始影象",1);
    cvNamedWindow("非畸變影象",1);
 
    show_colie = cvLoadImage(argv[1],-1);
    IplImage * clone=cvCloneImage(show_colie);
    cvShowImage("原始影象",show_colie);
    cvRemap(clone,show_colie,mapx,mapy);
    cvReleaseImage(&clone);
    cvShowImage("非畸變影象",show_colie);
</source lang>
 
   替換為:
<source lang="cpp" line>
    VideoCapture capture(argv[1]);
    if (!capture.isOpened()){
        return 0;
    }
    while(1){
        if (!capture.read(frame)){
            break;
        }
        ipFrame = frame;
        IplImage * clone=cvCloneImage(&ipFrame);
        cvShowImage("原始影象", &ipFrame);
        cvRemap(clone,show_colie,mapx,mapy);
        cvReleaseImage(&clone);
        cvShowImage("非畸變影象", show_colie);
 
        if (waitKey(5) == 'q'){
            break;
        }
    }
   就是將被校正的影象修改為,從video中獲取,迴圈校正顯示。

效果演示

  對應的圖片畸變校正效果如下:
  
                   原影象                                      校正後影象
程式碼下載:http://download.csdn.net/detail/u011630458/9268829