文章目录
前言
当今手机拍照盛行,尤其是手机拍摄一些文本资料,由于拍摄角度不同,拍摄出来的图像会有旋转角度,如果不进行矫正,对用户阅读文本不太方便。
1. 旋转文本矫正方法1主要步骤与C++实现
先上图(引用https://cloud.tencent.com/developer/article/1084455
这里介绍旋转文本矫正方法1的步骤:
- 读取图像并转换为灰度图像;
- 高斯模糊去除高频噪声的干扰;
- Canny边缘检测出文本信息;
- 对边缘二值图像进行kernel=(3,6)膨胀操作,迭代2次;
- 查找文本轮廓;
- 对每一文本轮廓获取角度信息(minAreaRect)
- 对所有轮廓的角度信息从小到大排序,获取中值角度;
- 根据中值角度值,使用getRotationMatrix2D获取旋转参数矩阵并应用投影变换warpAffine对图像进行旋转矫正。
C++实现:
#include <iostream>
#include <opencv2\imgcodecs.hpp>
#include <opencv2\core.hpp>
#include <opencv2\imgproc.hpp>
#include <opencv2\highgui.hpp>
#include <vector>
int main()
{
using namespace cv;
std::string strImgFile = "C:\\Temp\\common\\Workspace\\Opencv\\images\\rotated_text.jpg";
Mat mSrc = imread(strImgFile);
CV_Assert(!mSrc.empty());
imshow("src", mSrc);
Mat mGrey;
cvtColor(mSrc, mGrey, COLOR_BGR2GRAY);
CV_Assert(!mGrey.empty());
imshow("grey", mGrey);
Mat mGaussian;
GaussianBlur(mGrey, mGaussian, Size(5, 5), 1.0);
CV_Assert(!mGaussian.empty());
imshow("gaussian", mGaussian);
Mat mCanny;
Canny(mGaussian, mCanny, 20, 100);
CV_Assert(!mCanny.empty());
imshow("canny", mCanny);
Mat kernel = getStructuringElement(MORPH_RECT, Size(6, 3));
Mat mDialate;
dilate(mCanny, mDialate, kernel, Point(-1, -1), 2);
CV_Assert(!mDialate.empty());
imshow("dialate", mDialate);
std::vector<std::vector<Point>> contours;
findContours(mDialate, contours, RETR_EXTERNAL, CHAIN_APPROX_NONE);
std::vector<float> vecAngles;
for (int i = 0; i < contours.size(); i++)
{
RotatedRect rr = minAreaRect(contours[i]);
vecAngles.push_back(rr.angle);
}
std::sort(vecAngles.begin(), vecAngles.end());
for (auto angle : vecAngles)
{
std::cout << angle << " ";
}
std::cout << std::endl;
float angle = vecAngles[int(vecAngles.size() / 2) - 1];
std::cout << "angle:" << angle << std::endl;
Mat mResult;
Mat rotateM = getRotationMatrix2D(Point2f(mGrey.cols / 2.0, mGrey.rows / 2.0), angle, 1.0);
warpAffine(mSrc, mResult, rotateM, mGrey.size(), INTER_LINEAR, BORDER_CONSTANT, Scalar(255, 255, 255));
CV_Assert(!mResult.empty());
imshow("result", mResult);
waitKey(0);
destroyAllWindows();
system("pause");
return 0;
}
2. 结果展示
总结
这里只是给出了一种简单的实现方法,其实还有很多方法可以实现,比如霍夫线检测获取旋转角度,频域获取旋转角度等,后面还会一一尝试,谢谢大家!