c++ - 鱼眼失真校正

我在此链接http://www.tannerhelland.com/4743/simple-algorithm-correcting-lens-distortion/上看到了一种消除鱼眼镜头变形的算法，我试图用opencv在C++中实现它。当参数强度接近零时，输出图像与输入图像完全相同，并且使用更高的值会得到不好的结果。如果有人知道可能是什么问题(在我的代码中或更普遍地在算法中)，那将非常有帮助。
非常感谢你。

#include "opencv2\core\core.hpp"
#include "opencv2\highgui\highgui.hpp"
#include "opencv2\calib3d\calib3d.hpp"
#include <stdio.h>
#include <iostream>
#include <math.h>

using namespace std;
using namespace cv;

int main() {

    cout << " Usage: display_image ImageToLoadAndDisplay" << endl;
    Mat_<Vec3b> eiffel;
    eiffel = imread("C:/Users/Administrator/Downloads/TestFisheye.jpg", CV_LOAD_IMAGE_COLOR);   // Read the file
    if (!eiffel.data)                              // Check for invalid input
    {
        cout << "Could not open or find the image" << endl;
        return -1;
    }
    cout << "Input image depth: " << eiffel.depth() << endl;

    namedWindow("Display window", WINDOW_AUTOSIZE);// Create a window for display.
    imshow("Display window", eiffel);                   // Show our image inside it.

    //waitKey(0);                                          // Wait for a keystroke in the window

    int halfWidth = eiffel.rows / 2;
    int halfHeight = eiffel.cols / 2;
    double strength = 0.0001;
    double correctionRadius = sqrt(pow(eiffel.rows, 2) + pow(eiffel.cols, 2)) / strength;
    Mat_<Vec3b> dstImage = eiffel;

    int newX, newY;
    double distance;
    double theta;
    int sourceX;
    int sourceY;
    double r;
    for (int i = 0; i < dstImage.rows; ++i)
    {
        for (int j = 0; j < dstImage.cols; j++)
        {
            newX = i - halfWidth;
            newY = j - halfHeight;
            distance = sqrt(pow(newX, 2) + pow(newY, 2));
            r = distance / correctionRadius;
            if (r == 0.0)
                theta = 1;
            else
                theta = atan(r) / r;

            sourceX = round(halfWidth + theta*newX);
            sourceY = round(halfHeight + theta * newY);

            dstImage(i, j)[0] = eiffel(sourceX, sourceY)[0];
            dstImage(i, j)[1] = eiffel(sourceX, sourceY)[1];
            dstImage(i, j)[2] = eiffel(sourceX, sourceY)[2];
        }
    }

    namedWindow("Display window 2", WINDOW_AUTOSIZE);
    imshow("Display window 2", dstImage);                   // Show our image inside it.
    waitKey(0);

    return 0;
}

PS:我目前正在处理链接中发布的第一张图片。

最佳答案

您在这里有2个问题:

1-您需要将强度从0.0001增加到更合理的水平(尝试5)。

2-您使用相同的起点和终点矩阵。此Mat_<Vec3b> dstImage = eiffel;通常不会分配任何新内存。 dstImage只是指向原始图像的智能指针。因此，当您修改它时，您正在同时修改源图像。这将给您非常差的结果。改为Mat_<Vec3b> dstImage = eiffel.clone()
通过这些更改，我得到以下图像:


不太好，但至少快速且简单。

关于c++ - 鱼眼失真校正，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/35410744/