实现鱼眼矫正 C++ 的demo

以下是一个简单的 C++ 实现示例，实现对鱼眼图像的矫正：

#include <opencv2/opencv.hpp>

int main()
{
    // 加载鱼眼图像
    cv::Mat fisheyeImage = cv::imread("fisheye.jpg");

    // 定义矫正后图像的大小
    cv::Size imageSize(fisheyeImage.cols, fisheyeImage.rows);

    // 定义鱼眼矫正的参数
    float k = 0.5f; // 系数
    float cx = fisheyeImage.cols / 2.0f; // 中心点横坐标
    float cy = fisheyeImage.rows / 2.0f; // 中心点纵坐标

    // 计算鱼眼矫正后的图像
    cv::Mat undistortImage(imageSize, CV_8UC3);
    for (int y = 0; y < imageSize.height; ++y) {
        for (int x = 0; x < imageSize.width; ++x) {
            float dx = x - cx;
            float dy = y - cy;
            float r = sqrt(dx * dx + dy * dy);
            float theta = atan2(dy, dx);
            float r2 = r * r / fisheyeImage.cols;
            float x2 = r2 * cos(theta) + cx;
            float y2 = r2 * sin(theta) + cy;
            cv::Point2f pt(x2, y2);
            cv::Vec3b color = cv::Vec3b(0, 0, 0);
            if (x2 >= 0 && x2 < fisheyeImage.cols && y2 >= 0 && y2 < fisheyeImage.rows) {
                color = fisheyeImage.at<cv::Vec3b>(pt);
            }
            undistortImage.at<cv::Vec3b>(cv::Point(x, y)) = color;
        }
    }

    // 显示矫正前后的图像
    cv::namedWindow("Fisheye Lens", cv::WINDOW_NORMAL);
    cv::imshow("Fisheye Lens", fisheyeImage);
    cv::namedWindow("Undistort Image", cv::WINDOW_NORMAL);
    cv::imshow("Undistort Image", undistortImage);

    // 等待按键退出
    cv::waitKey(0);

    return 0;
}

在上述示例中，我们使用 OpenCV 库实现了鱼眼矫正，计算矫正后的图像并显示矫正前后的图像。在计算矫正后的图像时，我们遍历每个像素点，计算经过鱼眼矫正后的坐标，并从鱼眼图像中获取对应的颜色值。最后，我们使用 OpenCV 的 `namedWindow` 和 `imshow` 函数显示矫正前后的图像，并使用 `waitKey` 函数等待按键退出。