圖像畸變矯正

//
// Created by Qian.
//

#include <opencv2/opencv.hpp>
#include <string>

using namespace std;

string image_file = "../test.png";   // 請確保路徑正確

int main(int argc, char **argv) {

    // 本程序需要你自己實現去畸變部分的代碼。盡管我們可以調用OpenCV的去畸變，但自己實現一遍有助於理解。
    // 畸變參數
    double k1 = -0.28340811, k2 = 0.07395907, p1 = 0.00019359, p2 = 1.76187114e-05;
    // 內參
    double fx = 458.654, fy = 457.296, cx = 367.215, cy = 248.375;

    cv::Mat image = cv::imread(image_file,CV_8UC1);   // 圖像是灰度圖，CV_8UC1
    int rows = image.rows, cols = image.cols;
    cv::Mat image_undistort = cv::Mat(rows, cols, CV_8UC1);   // 去畸變以后的圖

    // 計算去畸變后圖像的內容
    for (int v = 0; v < rows; v++)
        for (int u = 0; u < cols; u++) {

            double u_distorted = 0, v_distorted = 0;
            // TODO 按照公式，計算點(u,v)對應到畸變圖像中的坐標(u_distorted, v_distorted) (~6 lines)
            // start your code here
            //首先轉化成歸一化坐標
            double x=(u-cx)/fx;
            double y=(v-cy)/fy;
            double r2=x*x+y*y;

            double x_distorted = x * (1 + k1 * r2 + k2 * r2 * r2)+2*p1*x*y+p2*(r2+2*x*x);
            double y_distorted= y * (1 + k1 * r2 + k2 * r2 * r2)+p1*(r2+2*y*y)+2*p2*x*y;
//            double x_distorted = x * (1 + k1 * r2 + k2 * r2 * r2);
//            double y_distorted= y * (1 + k1 * r2 + k2 * r2 * r2);

            //還原為像素坐標
            u_distorted = fx * x_distorted + cx;
            v_distorted = fy * y_distorted + cy;
            // end your code here

            // 賦值 (最近鄰插值)
            if (u_distorted >= 0 && v_distorted >= 0 && u_distorted < cols && v_distorted < rows) {
                image_undistort.at<uchar>(v, u) = image.at<uchar>((int) v_distorted, (int) u_distorted);
            } else {
                image_undistort.at<uchar>(v, u) = 0;
            }
        }

    // 畫圖去畸變后圖像
    cv::imshow("image undistorted", image_undistort);
    cv::waitKey();
}

    return 0;
}

---