| RMVL 2.3.0 Robotic Manipulation and Vision Library

载入中...
搜索中...
未找到
samples/camera/mv/sample_mv_manual_calib.cpp

迈德威视相机手动标定例程

#include <cstdio>
#include <opencv2/calib3d.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include "rmvl/algorithm/math.hpp"
#include "rmvl/core/timer.hpp"
#include "rmvl/camera/mv_camera.h"
using namespace std::chrono_literals;
// 内参矩阵
static cv::Matx33d cameraMatrix = {1250, 0, 640,
0, 1250, 512,
0, 0, 1};
// 畸变系数
static cv::Matx51d distCoeffs = {0, 0, 0, 0, 0};
static rm::MvCamera capture(rm::CameraConfig::create(rm::GrabMode::Continuous, rm::RetrieveMode::OpenCV));
void cameraMatrixCallBack(int pos, void *mat_pos_)
{
cv::Point *mat_pos = reinterpret_cast<cv::Point *>(mat_pos_);
// 内参矩阵
cameraMatrix(mat_pos->x, mat_pos->y) = pos;
}
void distCoeffCallBack(int pos, void *mat_pos_)
{
cv::Point *mat_pos = reinterpret_cast<cv::Point *>(mat_pos_);
// 畸变系数
if (mat_pos->x == 0 || mat_pos->x == 1)
distCoeffs(mat_pos->x, mat_pos->y) = static_cast<double>(-5000. + pos) / 5000.;
if (mat_pos->x == 2 || mat_pos->x == 3 || mat_pos->x == 4)
distCoeffs(mat_pos->x, mat_pos->y) = static_cast<double>(-500. + pos) / 5000.;
}
int main()
{
// 读取相机内参、畸变系数
const char *file_name = "out_calibration.yml";
cv::FileStorage fs_mv_in(file_name, cv::FileStorage::READ);
fs_mv_in["cameraMatrix"].isNone() ? void(0) : (fs_mv_in["cameraMatrix"] >> cameraMatrix);
fs_mv_in["distCoeffs"].isNone() ? void(0) : (fs_mv_in["distCoeffs"] >> distCoeffs);
int exposure = 10000;
int gain = 64;
int r_gain = 100;
int g_gain = 100;
int b_gain = 100;
// 设置相机参数
cv::FileStorage fs_mv_set("out_para.yml", cv::FileStorage::READ);
if (fs_mv_set.isOpened())
{
fs_mv_set["exposure"].isNone() ? void(0) : (fs_mv_set["exposure"] >> exposure);
fs_mv_set["gain"].isNone() ? void(0) : (fs_mv_set["gain"] >> gain);
fs_mv_set["r_gain"].isNone() ? void(0) : (fs_mv_set["r_gain"] >> r_gain);
fs_mv_set["g_gain"].isNone() ? void(0) : (fs_mv_set["g_gain"] >> g_gain);
fs_mv_set["b_gain"].isNone() ? void(0) : (fs_mv_set["b_gain"] >> b_gain);
}
capture.set(rm::CAMERA_MANUAL_EXPOSURE);
capture.set(rm::CAMERA_EXPOSURE, exposure);
capture.set(rm::CAMERA_GAIN, gain);
capture.set(rm::CAMERA_MANUAL_WB);
capture.set(rm::CAMERA_WB_RGAIN, r_gain);
capture.set(rm::CAMERA_WB_GGAIN, g_gain);
capture.set(rm::CAMERA_WB_BGAIN, b_gain);
namedWindow("图像画面", cv::WINDOW_NORMAL);
namedWindow("控制面板", cv::WINDOW_AUTOSIZE);
cv::Mat track_bar_img = cv::Mat::zeros(cv::Size(800, 1), CV_8UC1);
std::vector<cv::Point> matrix_pose; // 设置内参矩阵对应的参数位置
std::vector<cv::Point> dist_pose; // 设置畸变系数对应的参数位置
for (int i = 0; i < 3; i++)
for (int j = 0; j < 3; j++)
matrix_pose.emplace_back(i, j);
for (int i = 0; i < 6; i++)
dist_pose.emplace_back(i, 0);
// 设置滑动条
cv::createTrackbar("内参 (0, 0)", "控制面板", nullptr, 10000, cameraMatrixCallBack, &(matrix_pose.at(0)));
cv::setTrackbarPos("内参 (0, 0)", "控制面板", static_cast<int>(cameraMatrix(0, 0)));
cv::createTrackbar("内参 (1, 1)", "控制面板", nullptr, 10000, cameraMatrixCallBack, &(matrix_pose.at(0)));
cv::setTrackbarPos("内参 (1, 1)", "控制面板", static_cast<int>(cameraMatrix(1, 1)));
cv::createTrackbar("内参 (0, 2)", "控制面板", nullptr, 3000, cameraMatrixCallBack, &(matrix_pose.at(2)));
cv::setTrackbarPos("内参 (0, 2)", "控制面板", static_cast<int>(cameraMatrix(0, 2)));
cv::createTrackbar("内参 (1, 2)", "控制面板", nullptr, 3000, cameraMatrixCallBack, &(matrix_pose.at(5)));
cv::setTrackbarPos("内参 (1, 2)", "控制面板", static_cast<int>(cameraMatrix(1, 2)));
cv::createTrackbar("畸变 0", "控制面板", nullptr, 10000, distCoeffCallBack, &(dist_pose.at(0)));
cv::setTrackbarPos("畸变 0", "控制面板", static_cast<int>(distCoeffs(0, 0)) * 5000 + 5000);
cv::createTrackbar("畸变 1", "控制面板", nullptr, 10000, distCoeffCallBack, &(dist_pose.at(1)));
cv::setTrackbarPos("畸变 1", "控制面板", static_cast<int>(distCoeffs(1, 0)) * 5000 + 5000);
cv::createTrackbar("畸变 2", "控制面板", nullptr, 1000, distCoeffCallBack, &(dist_pose.at(2)));
cv::setTrackbarPos("畸变 2", "控制面板", static_cast<int>(distCoeffs(2, 0)) * 5000 + 500);
cv::createTrackbar("畸变 3", "控制面板", nullptr, 1000, distCoeffCallBack, &(dist_pose.at(3)));
cv::setTrackbarPos("畸变 3", "控制面板", static_cast<int>(distCoeffs(3, 0)) * 5000 + 500);
cv::createTrackbar("畸变 4", "控制面板", nullptr, 1000, distCoeffCallBack, &(dist_pose.at(4)));
cv::setTrackbarPos("畸变 4", "控制面板", static_cast<int>(distCoeffs(4, 0)) * 5000 + 500);
rm::Timer::sleep_for(1000);
printf("Press the 's' key to save the parameters to the yaml file: \033[33m%s\033[0m\n", file_name);
cv::Mat frame;
if (!capture.read(frame))
return -1;
resizeWindow("图像画面", cv::Size(static_cast<int>(frame.cols * 0.8),
static_cast<int>(frame.rows * 0.8)));
while (true)
{
if (!capture.read(frame))
continue;
if (frame.type() == CV_8UC1)
cv::cvtColor(frame, frame, cv::COLOR_GRAY2BGR);
// 图像矫正
cv::Mat map1, map2;
initUndistortRectifyMap(cameraMatrix, distCoeffs, cv::Mat(), cameraMatrix, frame.size(), CV_32FC1, map1, map2);
remap(frame, frame, map1, map2, cv::INTER_NEAREST);
// 绘制参照线
for (int i = 0; i <= frame.cols; i += frame.cols / 10)
line(frame, cv::Point(i, 0), cv::Point(i, frame.rows), cv::Scalar(0, 0, 255), 1);
for (int j = 0; j <= frame.rows; j += frame.rows / 10)
line(frame, cv::Point(0, j), cv::Point(frame.cols, j), cv::Scalar(0, 0, 255), 1);
imshow("图像画面", frame);
imshow("控制面板", track_bar_img);
char c = cv::waitKey(1);
// 退出程序
if (c == 27)
{
if (cv::waitKey(0) == 27)
break;
}
// 保存参数
else if (c == 's')
{
cv::FileStorage fs_mv_out(file_name, cv::FileStorage::WRITE);
fs_mv_out << "cameraMatrix" << cameraMatrix;
fs_mv_out << "distCoeffs" << distCoeffs;
printf("\033[32mSuccess to write the parameters into \"%s\"\033[0m\n", file_name);
printf(" ┌ %-5.4g, %-5.4g, %-5.4g ┐\n"
" -- cameraMatrix: │ %-5.4g, %-5.4g, %-5.4g │\n"
" └ %-5.4g, %-5.4g, %-5.4g ┘\n",
cameraMatrix(0, 0), cameraMatrix(0, 1), cameraMatrix(0, 2),
cameraMatrix(1, 0), cameraMatrix(1, 1), cameraMatrix(1, 2),
cameraMatrix(2, 0), cameraMatrix(2, 1), cameraMatrix(2, 2));
printf(" ┌ %-8.5g ┐\n"
" │ %-8.5g │\n"
" -- distCoeffs: │ %-8.5g │\n"
" │ %-8.5g │\n"
" └ %-8.5g ┘\n",
distCoeffs(0), distCoeffs(1), distCoeffs(2), distCoeffs(3), distCoeffs(4));
}
}
return 0;
}
mv_camera.h
迈德威视相机库
rm::MvCamera
迈德威视相机库
定义 mv_camera.h:38
rm::Timer::sleep_for
static void sleep_for(double t)
当前线程休眠指定时间
cv::Matx51d
Matx< double, 5, 1 > Matx51d
定义 math.hpp:72
rm::RetrieveMode::OpenCV
@ OpenCV
使用 OpenCV 的 'cvtColor' 进行处理
定义 camutils.hpp:54
rm::CAMERA_WB_BGAIN
@ CAMERA_WB_BGAIN
白平衡蓝色分量
定义 camutils.hpp:108
rm::CAMERA_WB_GGAIN
@ CAMERA_WB_GGAIN
白平衡绿色分量
定义 camutils.hpp:107
rm::CAMERA_MANUAL_EXPOSURE
@ CAMERA_MANUAL_EXPOSURE
手动曝光
定义 camutils.hpp:99
rm::CAMERA_EXPOSURE
@ CAMERA_EXPOSURE
曝光值
定义 camutils.hpp:103
rm::CAMERA_MANUAL_WB
@ CAMERA_MANUAL_WB
手动白平衡
定义 camutils.hpp:102
rm::CAMERA_WB_RGAIN
@ CAMERA_WB_RGAIN
白平衡红色分量
定义 camutils.hpp:106
rm::CAMERA_GAIN
@ CAMERA_GAIN
模拟增益
定义 camutils.hpp:104
rm::GrabMode::Continuous
@ Continuous
连续采样
定义 camutils.hpp:36
math.hpp
基础数学库
rm::CameraConfig::create
static CameraConfig create(Args &&...modes)
创建相机初始化配置模式
定义 camutils.hpp:72
timer.hpp
定时、计时模块