导读
在做OCR票据类识别的时候经常会遇到一些票据上会有印章 , 而对于的文字检测和文字识别模型而言,印章的存在一定会影响模型识别的准确率 , 所以通常我们都是先将图片去除印章之后,再将图片送入到文字检测和文字识别模型中 。本篇文章就介绍一个比较简单的方法用来去除红色印章
移除红色印章
我们通过分离图片的通道,提取图片的红色通道,然后再通过阈值来去除红色的印章import cv2import numpy as npdef remove_red_seal(input_img):# 分离图片的通道blue_c, green_c, red_c = cv2.split(input_img)#利用大津法自动选择阈值thresh, ret = cv2.threshold(red_c, 0, 255,cv2.THRESH_OTSU)#对阈值进行调整filter_condition = int(thresh * 0.90)#移除红色的印章_, red_thresh = cv2.threshold(red_c, filter_condition, 255, cv2.THRESH_BINARY)# 把图片转回3通道result_img = np.expand_dims(red_thresh, axis=2)result_img = np.concatenate((result_img, result_img, result_img), axis=-1)return result_imginput_img = cv2.imread("1.jpg")remove_seal = remove_red_seal(input_img)cv2.imwrite("remove_seal.jpg",remove_seal)
去除印章前
【一 OCR图片预处理之去除印章】去除印章后
注意:对于不同的场景,你可能需要对阈值进行微调(百分比) , 以获取你认为的最佳阈值,百分比越小红色印章移除的越干净 , 同时也有可能会移除部分文字信息 。
threshold函数
threshold(src, thresh, maxval, type[, dst])->ret,dst- src::灰度图或单通道图片
- thresh:阈值
- maxval:最大值
- type:阈值类型
在对图片做二值化处理的时候需要设置一个阈值来对图片进行二值化处理,然而在部分复杂的场景下,如果采用固定的阈值可能在某些场景下效果不错,换到有些场景时效果就不行了 。
这时候我们就会想要用自动的阈值,这时候就可以用到THRESH_OTSU和THRESH_TRIANGLE这两个参数,它们会根据图片的灰度直方图来计算出一个阈值将图片分为前景和背景,下面我们来介绍一下它们是如何实现的 。
THRESH_OTSU
大津法(OTSU):也被称为是最大类间差法,被认为是图像分割阈值选择的最佳算法,计算简单,鲁棒性较好 , 不受图像亮度和和对比度的影响 , 因此在数字图像处理中被广泛的使用 。它根据图像的灰度直方图,将图像分为前景和背景两个部分 。因为方差是度量图像灰度分布是否均匀,如果图像的背景和前景之间的差别越大,那么它们之间的类间方差差距也会越大 。所以,如果我们能够保证图像前景和背景的灰度直方图方差差距最大时,就能让前景和背景分离的效果达到最佳 , 实际效果还是取决于具体的场景 , 可能需要根据不同的需求对阈值进行微调 。
- 公式推导
假设灰度T是图像分割前景和背景的最佳阈值,图像上任意一点属于前景的概率为ω1,属于背景的概率为ω2 。图像前景的平均灰度值为μ1?,背景的平均灰度值为μ2?,所以图像的平均灰度值μ为
根据类间的方差计算公式,前景和背景的类间方差计算如下
因为
结合上面3个式子可得
为了方便我们后面编程来实现,还需要对上式做一些调整 , 这里引入几个参数pi表示灰度值等于i的概率,图像的灰度取值在[ 0 , 255 ] 范围内取整数 。假设灰度值t可以使图像前景和背景的方差最大 , m1?为灰度级t t