python - 使用 OpenCV 和 Python 去歪斜文本(RotatedRect、minAreaRect)

Question

我是 OpenCV 的新手，我想对具有倾斜文本的图像进行校正。首先我读取 GrayScale 中的图像并将其二值化，然后我尝试做 this :

import cv2
import numpy as np

img = cv2.imread('m20.jpg',0)
ret,byw = cv2.threshold(img,127,255,cv2.THRESH_BINARY_INV)
_, contours, hierarchy = cv2.findContours(byw.copy(), cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
cnt = contours[0]
draw = cv2.cvtColor(byw, cv2.COLOR_GRAY2BGR)
rect = cv2.minAreaRect(cnt)
box = cv2.boxPoints(rect)
box = np.int0(box)
cv2.drawContours(draw, [box], 0, (0, 255, 0), 2)

但不起作用，因为 findContours() 期望接收具有 body 形状的图像。我尝试的另一种方法是翻译这段 C++ 代码:

// Read image
Mat3b img = imread("path_to_image");

// Binarize image. Text is white, background is black
Mat1b bin;
cvtColor(img, bin, COLOR_BGR2GRAY);

bin = bin < 200;

// Find all white pixels
vector<Point> pts;
findNonZero(bin, pts);

// Get rotated rect of white pixels
RotatedRect box = minAreaRect(pts);
if (box.size.width > box.size.height)
{
    swap(box.size.width, box.size.height);
    box.angle += 90.f;
}

Point2f vertices[4];
box.points(vertices);

for (int i = 0; i < 4; ++i)
{
    line(img, vertices[i], vertices[(i + 1) % 4], Scalar(0, 255, 0));
}

// Rotate the image according to the found angle
Mat1b rotated;
Mat M = getRotationMatrix2D(box.center, box.angle, 1.0);
warpAffine(bin, rotated, M, bin.size());

我有这个:

draw = cv2.cvtColor(byw, cv2.COLOR_GRAY2BGR)
data = np.array(byw)
subzero = np.nonzero(data)
subuno = np.reshape(subzero,(17345,2)) # this is because cv2.minAreaRect() receives a Nx2 numpy
rect = cv2.minAreaRect(subuno)
box = cv2.boxPoints(rect)
box = np.int0(box)
cv2.drawContours(draw,[box],0,(0,255,0),2)

但结果又不是预期的，矩形没有很好地定位。

---

我也想到可能会尝试像在 C++ 中那样制作 for 但我不知道如何从 box = cv2.boxPoints(rect)。请帮忙!

Answer 1

也许你可以看看这个:http://www.pyimagesearch.com/2014/08/25/4-point-opencv-getperspective-transform-example/

在该链接中，作者对整个文档(以及包含的文本)进行了校正或转换，但是，这取决于根据图像中找到的轮廓找到文档的边缘。

他在以下教程中更进一步:http://www.pyimagesearch.com/2014/09/01/build-kick-ass-mobile-document-scanner-just-5-minutes/

他的解决方案之所以奏效，是因为他可以根据检测到的文档位置、方向和倾斜度调整整个文档。调整整个文档的位置实际上会调整文档中的所有内容，包括文本。

但是，我相信您要问的是您希望在不检测到任何文档边缘和轮廓的情况下对文本进行校正。如果是这种情况，那么我假设你需要提供另一个基础或标准来作为你的文本去偏斜的基础(即检测图像中有字母，然后根据你的标准检测字母的偏斜程度, 然后调整字母), 这可能是一个不平凡的练习。