From 5f06ddc4477bd9deec3040f7f6af9cdfeb2b9f09 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=A9=AC=E5=AD=A6=E6=B5=9C=20xuebang=5Fma?= Date: Sun, 29 Dec 2024 21:40:35 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=9B=BE=E5=83=8F=E7=95=B8?= =?UTF-8?q?=E5=8F=98=E7=9F=AB=E6=AD=A3=E5=92=8Cpixel=E5=AF=B9=E5=BA=94?= =?UTF-8?q?=E7=89=A9=E7=90=86=E5=A4=A7=E5=B0=8F=E8=AE=A1=E7=AE=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/PBBiology/include/PBImageProcess.h | 11 +++- src/PBBiology/src/PBImageProcess.cpp | 87 +++++++++++++++++++++++++- 2 files changed, 94 insertions(+), 4 deletions(-) diff --git a/src/PBBiology/include/PBImageProcess.h b/src/PBBiology/include/PBImageProcess.h index d64644a..0e1bffb 100644 --- a/src/PBBiology/include/PBImageProcess.h +++ b/src/PBBiology/include/PBImageProcess.h @@ -47,4 +47,13 @@ Mat get_magic_wand_image(Mat src,int x,int y,float max,float min); // 锐化 -Mat SetSharpen(Mat src); \ No newline at end of file +Mat SetSharpen(Mat src); +//相机标定功能，gray是包含完整棋盘格灰度图像，patternSize是棋盘格内角点的数量（假如棋盘格的尺寸是9*7，那内角点就是8*6），grid_size是每个格子的物理大小 +//cameraMatrix和distCoeffs是后续图像畸变矫正需要的参数，标定一次后，镜头无变动情况下，后续图像直接用此参数矫正就可以，pixel_size是矫正后每个pixel的物理大小 +//返回值是0表示标定错误，返回值是1表示标定正确 +//注：相机标定功能得到的结果参数只能用于矫正和计算得到结果的输入图像分辨率一致的图 +bool camera_calibration(Mat gray,cv::Size patternSize,float grid_size,cv::Mat& cameraMatrix,cv::Mat& distCoeffs,float& pixel_size); +//图像畸变矫正功能，image是输入图像（无要求），cameraMatrix和distCoeffs是图像畸变矫正的参数 +//返回值是矫正后的图像，输入图像是什么格式，输出图像就是什么格式 +//注：相机标定功能得到的结果参数只能用于矫正和计算得到结果的输入图像分辨率一致的图 +Mat distortion_correction(Mat image,cv::Mat cameraMatrix,cv::Mat distCoeffs); diff --git a/src/PBBiology/src/PBImageProcess.cpp b/src/PBBiology/src/PBImageProcess.cpp index e85cf97..3028ccf 100644 --- a/src/PBBiology/src/PBImageProcess.cpp +++ b/src/PBBiology/src/PBImageProcess.cpp @@ -1110,7 +1110,8 @@ Mat bgr_scale_image(Mat src, float maxVal, float minVal, int scientific_flag) float rotia = 255.0 / (maxVal - minVal); Mat image = Mat(h + 2 * start, 3 * w, CV_8UC3); image.setTo(255); - src.copyTo(image(Rect(0, start, w, h))); + src.copyTo(image(Rect(0,start,w,h))); + cv::rectangle(image, Rect(0,start,w,h), cv::Scalar(0,0,0), 2); float diff = maxVal - minVal; int exponent = static_cast(std::log10(diff)); @@ -1188,8 +1189,7 @@ Mat bgr_scale_image(Mat src, float maxVal, float minVal, int scientific_flag) std::ostringstream oss; if (tickData[i] == int(tickData[i])) { oss << static_cast(tickData[i]); - } - else { + } else { oss.precision(1); oss << std::fixed << tickData[i]; } @@ -1287,4 +1287,85 @@ Mat SetSharpen(Mat src) usm = src + 0.5 * (src - blur_img); return usm; +} +bool camera_calibration(Mat gray,cv::Size patternSize,float grid_size,cv::Mat& cameraMatrix,cv::Mat& distCoeffs,float& pixel_size) +{ + if(gray.type() != CV_8UC1) + { + std::cerr << "Error: gray not CV_8UC1!" << std::endl; + return 0; + } + std::vector corners; + bool found = cv::findChessboardCorners(gray, patternSize, corners, + cv::CALIB_CB_ADAPTIVE_THRESH | cv::CALIB_CB_NORMALIZE_IMAGE); + + if (found) { + cv::cornerSubPix(gray, corners, cv::Size(3, 3), cv::Size(-1, -1), + cv::TermCriteria(cv::TermCriteria::EPS | cv::TermCriteria::MAX_ITER, 30, 0.1)); + } else { + std::cerr << "Error: Chessboard corners not found!" << std::endl; + return 0; + } + std::vector> objectPoints; + std::vector> imagePoints; + + std::vector objPts; + for (int i = 0; i < patternSize.height; ++i) { + for (int j = 0; j < patternSize.width; ++j) { + objPts.push_back(cv::Point3f(j, i, 0.0f)); + } + } + objectPoints.push_back(objPts); + imagePoints.push_back(corners); + cameraMatrix = cv::Mat::eye(3, 3, CV_64F); // 相机矩阵初始化 + distCoeffs = cv::Mat::zeros(8, 1, CV_64F); // 畸变系数初始化 + std::vector rvecs, tvecs; + int calibrationFlags = 0; + TermCriteria criteria = TermCriteria(TermCriteria::MAX_ITER + TermCriteria::EPS, 30, DBL_EPSILON); + cv::calibrateCamera(objectPoints, imagePoints, gray.size(), cameraMatrix, distCoeffs, rvecs, tvecs,calibrationFlags,criteria); + + vector undistortedCorners; + undistortPoints(corners, undistortedCorners, cameraMatrix, distCoeffs); + + std::vector homogeneousPoints; + for (const auto& p : undistortedCorners) { + homogeneousPoints.push_back(cv::Point3f(p.x, p.y, 1.0f)); // 将 (x, y) 转换为 (x, y, 1) + } + std::vector pixelPoints; + for (const auto& p : homogeneousPoints) { + cv::Mat pointMat = (cv::Mat_(3, 1) << p.x, p.y, p.z); + cv::Mat transformedPoint = cameraMatrix * pointMat; + float x = transformedPoint.at(0, 0) / transformedPoint.at(2, 0); + float y = transformedPoint.at(1, 0) / transformedPoint.at(2, 0); + pixelPoints.push_back(cv::Point2f(x, y)); + } + int numCols = patternSize.width; + int numRows = patternSize.height; + float meanSideLength = 0; + int num = 0; + for (int i = 0; i < (patternSize.height - 1); ++i) { + for (int j = 0; j < (patternSize.width - 1); ++j) { + int index = i * patternSize.width + j; + if (j + 1 < numCols) { + float width = static_cast(norm(pixelPoints[index] - pixelPoints[index + 1])); + meanSideLength+=(width); + num++; + } + if (i + 1 < patternSize.height) { + float height = static_cast(norm(pixelPoints[index] - pixelPoints[index + numCols])); + meanSideLength+=(height); + num++; + } + } + } + meanSideLength /= num; + pixel_size = grid_size/meanSideLength; + return 1; +} + +Mat distortion_correction(Mat image,cv::Mat cameraMatrix,cv::Mat distCoeffs) +{ + cv::Mat undistortedImage; + cv::undistort(image, undistortedImage, cameraMatrix, distCoeffs); + return undistortedImage; } \ No newline at end of file