C++ OpenCV实战:旋转文本矫正方法1


前言

当今手机拍照盛行,尤其是手机拍摄一些文本资料,由于拍摄角度不同,拍摄出来的图像会有旋转角度,如果不进行矫正,对用户阅读文本不太方便。


1. 旋转文本矫正方法1主要步骤与C++实现

先上图(引用https://cloud.tencent.com/developer/article/1084455
请添加图片描述
这里介绍旋转文本矫正方法1的步骤:

  1. 读取图像并转换为灰度图像;
  2. 高斯模糊去除高频噪声的干扰;
  3. Canny边缘检测出文本信息;
  4. 对边缘二值图像进行kernel=(3,6)膨胀操作,迭代2次;
  5. 查找文本轮廓;
  6. 对每一文本轮廓获取角度信息(minAreaRect)
  7. 对所有轮廓的角度信息从小到大排序,获取中值角度;
  8. 根据中值角度值,使用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. 结果展示

在这里插入图片描述

总结

这里只是给出了一种简单的实现方法,其实还有很多方法可以实现,比如霍夫线检测获取旋转角度,频域获取旋转角度等,后面还会一一尝试,谢谢大家!

参考

https://cloud.tencent.com/developer/article/1084455

Guess you like

Origin blog.csdn.net/DU_YULIN/article/details/120504660