u2net pytorch提取印章 提取图片中印章

您所在的位置:网站首页 文件提取印章怎么操作的 u2net pytorch提取印章 提取图片中印章

u2net pytorch提取印章 提取图片中印章

2024-07-12 17:43| 来源: 网络整理| 查看: 265

导读

在做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)

u2net pytorch提取印章 提取图片中印章_方差

去除印章前

u2net pytorch提取印章 提取图片中印章_灰度值_02

去除印章后

注意:对于不同的场景,你可能需要对阈值进行微调(百分比),以获取你认为的最佳阈值,百分比越小红色印章移除的越干净,同时也有可能会移除部分文字信息。

threshold函数

threshold(src, thresh, maxval, type[, dst])->ret,dst

src::灰度图或单通道图片thresh:阈值maxval:最大值type:阈值类型

这里重点介绍一下type参数的取值,它的取值如下图所示

u2net pytorch提取印章 提取图片中印章_方差_03

在对图片做二值化处理的时候需要设置一个阈值来对图片进行二值化处理,然而在部分复杂的场景下,如果采用固定的阈值可能在某些场景下效果不错,换到有些场景时效果就不行了。

这时候我们就会想要用自动的阈值,这时候就可以用到THRESH_OTSU和THRESH_TRIANGLE这两个参数,它们会根据图片的灰度直方图来计算出一个阈值将图片分为前景和背景,下面我们来介绍一下它们是如何实现的。

THRESH_OTSU

大津法(OTSU):也被称为是最大类间差法,被认为是图像分割阈值选择的最佳算法,计算简单,鲁棒性较好,不受图像亮度和和对比度的影响,因此在数字图像处理中被广泛的使用。

它根据图像的灰度直方图,将图像分为前景和背景两个部分。因为方差是度量图像灰度分布是否均匀,如果图像的背景和前景之间的差别越大,那么它们之间的类间方差差距也会越大。所以,如果我们能够保证图像前景和背景的灰度直方图方差差距最大时,就能让前景和背景分离的效果达到最佳,实际效果还是取决于具体的场景,可能需要根据不同的需求对阈值进行微调。

公式推导

其实只要抓住大津法的核心思想最大化前景和背景的方差要推导公式用代码来实现并不难,接下来我们来推导一下这个公式。

假设灰度T是图像分割前景和背景的最佳阈值,图像上任意一点属于前景的概率为ω1,属于背景的概率为ω2。图像前景的平均灰度值为μ1,背景的平均灰度值为μ2,所以图像的平均灰度值μ为

u2net pytorch提取印章 提取图片中印章_灰度直方图_04

根据类间的方差计算公式,前景和背景的类间方差计算如下

u2net pytorch提取印章 提取图片中印章_二值化图像去除小黑点_05

因为

u2net pytorch提取印章 提取图片中印章_灰度直方图_06

结合上面3个式子可得

u2net pytorch提取印章 提取图片中印章_二值化图像去除小黑点_07

为了方便我们后面编程来实现,还需要对上式做一些调整,这里引入几个参数pi表示灰度值等于i的概率,图像的灰度取值在[ 0 , 255 ] 范围内取整数。假设灰度值t可以使图像前景和背景的方差最大,m1为灰度级t tt的累加均值,m为图像的灰度级L的均值累加

u2net pytorch提取印章 提取图片中印章_方差_08

可得μ1和μ2

u2net pytorch提取印章 提取图片中印章_灰度直方图_09

接下来我们对δ2 结合上面的式子做个变换

u2net pytorch提取印章 提取图片中印章_u2net pytorch提取印章_10

我们只需要使上式最大化即可

代码实现OTSU

上面我们推导了大津法的公式,以及如何来求解阈值划分前景和背景,下面我们用python来实现这个算法

u2net pytorch提取印章 提取图片中印章_u2net pytorch提取印章_11

大津法

比较一下我们自己实现的大津法和opencv内置的函数

u2net pytorch提取印章 提取图片中印章_二值化图像去除小黑点_12

最终两者输出的阈值都是160,不过python实现的代码是opencv时间的25倍左右,所以python在这方面对比c确实是硬伤。

THRESH_TRIANGLE

三角法(TRIANGLE):是基于直方图利用几何的方法来求分割的最佳阈值,假设的成立条件是直方图的最大波峰在靠近最亮的一侧,然后再通过三角形来求解最大的距离找到最佳阈值。

u2net pytorch提取印章 提取图片中印章_二值化图像去除小黑点_13

如图所示,在灰度直方图上,从最高峰bmax到最暗对应直方图的bmin构造一条直线,然后从bmin 到bmax开始计算到直线的垂直距离d,当d达到最大时,此时所对应的灰度值t就是分割图像的最佳阈值

接下来我们看看,使用三角法求解阈值值的整个流程,这里引入两个参数灰度级L和频率f:

将图片转换为灰度图,通过OpenCV可以很容易实现计算灰度图的灰度直方图,也就是每个灰度级L对应的频率f 对灰度直方图进行排序,按灰度级进行排序,由小到大确定直方图最大值(也就是f)所对应灰度级L的位置,如果在左侧(灰度值小)就需要对灰度直方图进行翻转根据左侧边界的灰度级点( Lmin , fmin)和最亮部分频率最大对应的灰度级点( Lmax , fmax) ,由两点式我们可以确定这条直线计算Lmin到Lmax的任意一点(L, f )到直线的距离d,当d最大时所对应的L就是我们要求的最佳阈值

u2net pytorch提取印章 提取图片中印章_u2net pytorch提取印章_14

代码实现

u2net pytorch提取印章 提取图片中印章_方差_15

大津法和三角法的对比共同点:两者都是算法自动计算出阈值,不需要指定阈值不同点:大津法适合双波峰的灰度直方图,三角法适合单波峰的灰度直方图opencv显示灰度直方图

u2net pytorch提取印章 提取图片中印章_方差_16

u2net pytorch提取印章 提取图片中印章_二值化图像去除小黑点_17

 



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3