写一段弯曲文本矫正python代码

以下是一段 Python 代码，用于对弯曲文本进行矫正：

from PIL import Image
import pytesseract
import cv2

# 读取图片
img = cv2.imread('curved_text.png')

# 灰度化处理
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 二值化处理
_, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)

# 边缘检测
edges = cv2.Canny(thresh, 50, 150, apertureSize=3)

# 膨胀操作
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (3, 3))
dilated = cv2.dilate(edges, kernel, iterations=1)

# 轮廓检测
contours, hierarchy = cv2.findContours(dilated, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# 获取最大的轮廓
max_contour = max(contours, key=cv2.contourArea)

# 获取最大轮廓的矩形边框
x, y, w, h = cv2.boundingRect(max_contour)

# 裁剪图片，只保留矩形区域
cropped = img[y:y+h, x:x+w]

# OCR 识别
text = pytesseract.image_to_string(Image.fromarray(cropped), lang='eng')

print(text)

这段代码使用了 OpenCV 和 pytesseract 库，通过对图片进行一系列处理，最终得到了矫正后的文本。但是，需要注意的是，该代码只适用于一定程度的弯曲文本矫正，对于过于弯曲或扭曲的文本可能效果不佳。