【opencv】图像畸变校正

接上篇文章:【鱼眼+普通相机】相机标定

附代码:

方法一:

使用cv2.undistort

python 复制代码
"""
    Create May 11, 2024
    @author Wang Jiajun
"""

import cv2
import numpy as np


def correct(img,camera_file='E:/calib_yaml/ship_calib.yaml'):
    """
    矫正图像,去除畸变
        img: 输入图像
        camera_file: 相机参数文件
        return: 矫正后的图像
    """
    fs = cv2.FileStorage(camera_file, cv2.FILE_STORAGE_READ)
    camera_matrix = fs.getNode("camera_matrix").mat()
    dist_coeffs = fs.getNode("dist_coeffs").mat()  # 畸变参数--顺序很重要!!!:k1,k2,p1,p2,k3
    resolution = fs.getNode("resolution").mat()  # 图像分辨率
    imSize = (resolution[0][0], resolution[1][0])
    fs.release()

    # 矫正图像
    img_distort = cv2.undistort(img, np.array(camera_matrix), np.array(dist_coeffs))
    return img_distort 


if __name__ == '__main__':
    camera_file = 'E:/calib_yaml/ship_calib.yaml'
    video = cv2.VideoCapture(0)
    while True:
        ret, img = video.read()
        # 3. 矫正图像
        img = cv2.resize(img, (640, 480))
        cv2.putText(img, "Press q to exit", (20, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (255, 200, 0), 2)
        img_distort = cv2.correct(img, camera_file )
        # 4. 显示原图和矫正后的图像
        cv2.imshow('img_original', img)
        cv2.imshow('img_distort', img_distort)
        key = cv2.waitKey(1)
        # 5. 按q退出
        if key & 0xFF == ord('q'):
            break
    # 6. 关闭窗口
    video.release()
    cv2.destroyAllWindows()

方法二:

使用cv2.initUndistortRectifyMap建立映射关系,就不用每次读取内参了,速度更快。

python 复制代码
"""
    Create May 11, 2024
    @author Wang Jiajun
"""

import cv2
import numpy as np


def correct(camera_file='E:/calib_yaml/ship_calib.yaml'):
    """
    矫正图像,去除畸变
        img: 输入图像
        camera_file: 相机参数文件
        return: 矫正后的图像
    """
    fs = cv2.FileStorage(camera_file, cv2.FILE_STORAGE_READ)
    camera_matrix = fs.getNode("camera_matrix").mat()
    dist_coeffs = fs.getNode("dist_coeffs").mat()  # 畸变参数--顺序很重要!!!:k1,k2,p1,p2,k3
    resolution = fs.getNode("resolution").mat()  # 图像分辨率
    imSize = (resolution[0][0], resolution[1][0])
    fs.release()

    # 矫正图像
    new_camera_matrix, roi = cv2.getOptimalNewCameraMatrix(np.array(camera_matrix), np.array(dist_coeffs),
                                                           imSize, 0,
                                                           imSize)
    map1, map2 = cv2.initUndistortRectifyMap(np.array(camera_matrix), np.array(dist_coeffs), None, new_camera_matrix,
                                             imSize, cv2.CV_32FC1)
    # img_distort = cv2.undistort(img, np.array(camera_matrix), np.array(dist_coeffs))
    return map1, map2


if __name__ == '__main__':
    camera_file = 'E:/calib_yaml/ship_calib.yaml'
    map1, map2 = correct(camera_file)
    video = cv2.VideoCapture(0)
    while True:
        ret, img = video.read()
        # 3. 矫正图像
        img = cv2.resize(img, (640, 480))
        cv2.putText(img, "Press q to exit", (20, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (255, 200, 0), 2)
        img_distort = cv2.remap(img, map1, map2, cv2.INTER_LINEAR)
        # 4. 显示原图和矫正后的图像
        cv2.imshow('img_original', img)
        cv2.imshow('img_distort', img_distort)
        key = cv2.waitKey(1)
        # 5. 按q退出
        if key & 0xFF == ord('q'):
            break
    # 6. 关闭窗口
    video.release()
    cv2.destroyAllWindows()
相关推荐
阿坡RPA8 小时前
手搓MCP客户端&服务端:从零到实战极速了解MCP是什么?
人工智能·aigc
用户27784491049938 小时前
借助DeepSeek智能生成测试用例:从提示词到Excel表格的全流程实践
人工智能·python
机器之心8 小时前
刚刚,DeepSeek公布推理时Scaling新论文,R2要来了?
人工智能
算AI10 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
凯子坚持 c11 小时前
基于飞桨框架3.0本地DeepSeek-R1蒸馏版部署实战
人工智能·paddlepaddle
你觉得20511 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义下载方法
大数据·人工智能·python·gpt·学习·机器学习·aigc
8K超高清11 小时前
中国8K摄像机:科技赋能文化传承新图景
大数据·人工智能·科技·物联网·智能硬件
hyshhhh12 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
薛定谔的猫-菜鸟程序员12 小时前
零基础玩转深度神经网络大模型:从Hello World到AI炼金术-详解版(含:Conda 全面使用指南)
人工智能·神经网络·dnn
币之互联万物12 小时前
2025 AI智能数字农业研讨会在苏州启幕,科技助农与数据兴业成焦点
人工智能·科技