Python验证码自动识别
先容
在python爬虫爬取某些网站的验证码的时候大概会碰着验证码识此外问题,本日就来看下如何让呆板自动识别验证码。
识别验证码凡是是这几个步调:
1、灰度处理惩罚
2、二值化
3、去除边框(假如有的话)
4、降噪
5、切割字符可能倾斜度改正
6、练习字体库
7、识别
这6个步调中前三个步调是根基的,4可能5可按照实际环境选择是否需要,并不必然切割验证码,识别率就会上升许多有时候还会下降
本日讲的不涉及练习字体库的内容,感乐趣的同学请自行搜索相关文章。
几个主要的验证码识别相关的python库: Pillow(python图像处理惩罚库)、OpenCV(高级图像处理惩罚库)、pytesseract(识别库)
灰度处理惩罚&二值化
灰度处理惩罚,就是把彩色的验证码图片转为灰色的图片。
二值化,是将图片处理惩罚为只有利害两色的图片,利于后头的图像处理惩罚和识别
在OpenCV中有现成的要领可以举办灰度处理惩罚和二值化,处理惩罚后的结果:
代码:
# 自适应阀值二值化 def _get_dynamic_binary_image(filedir, img_name): filename = './out_img/' + img_name.split('.')[0] + '-binary.jpg' img_name = filedir + '/' + img_name print('.....' + img_name) im = cv2.imread(img_name) im = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY) #灰值化 # 二值化 tp = cv2.adaptiveThreshold(im, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 21, 1) cv2.imwrite(filename,tp) return tp
去除边框
假如验证码有边框,那我们就需要去除边框,去除边框就是遍历像素点,找到四个边框上的所有点,把他们都改为白色,我这里边框是两个像素宽
留意:在用OpenCV时,图片的矩阵点是反的,就是长和宽是颠倒的
代码:
# 去除边框 def clear_border(img,img_name): filename = './out_img/' + img_name.split('.')[0] + '-clearBorder.jpg' h, w = img.shape[:2] for y in range(0, w): for x in range(0, h): if y < 2 or y > w - 2: img[x, y] = 255 if x < 2 or x > h -2: img[x, y] = 255 cv2.imwrite(filename,img) return img
结果:
降噪
降噪是验证码处理惩罚中较量重要的一个步调,我这里利用了点降噪和线降噪
线降噪的思路就是检测这个点相邻的四个点(图中标出的绿色点),判定这四个点中是白点的个数,假如有两个以上的白色像素点,那么就认为这个点是白色的,从而去除整个滋扰线,可是这种要领是有限度的,假如滋扰线出格粗就没有步伐去除,只能去除细的滋扰线
代码:
# 滋扰线降噪 def interference_line(img, img_name): filename = './out_img/' + img_name.split('.')[0] + '-interferenceline.jpg' h, w = img.shape[:2] # !!!opencv矩阵点是反的 # img[1,2] 1:图片的高度,2:图片的宽度 for y in range(1, w - 1): for x in range(1, h - 1): count = 0 if img[x, y - 1] > 245: count = count + 1 if img[x, y + 1] > 245: count = count + 1 if img[x - 1, y] > 245: count = count + 1 if img[x + 1, y] > 245: count = count + 1 if count > 2: img[x, y] = 255 cv2.imwrite(filename,img) return img
点降噪的思路和线降噪的差不多,只是会针对差异的位置检测的点纷歧样,注释写的很清楚了
#p#分页标题#e#
代码:
# 点降噪 def interference_point(img,img_name, x = 0, y = 0): """ 9邻域框,以当前点为中心的田字框,斑点个数 :param x: :param y: :return: """ filename = './out_img/' + img_name.split('.')[0] + '-interferencePoint.jpg' # todo 判定图片的长宽度下限 cur_pixel = img[x,y]# 当前像素点的值 height,width = img.shape[:2] for y in range(0, width - 1): for x in range(0, height - 1): if y == 0: # 第一行 if x == 0: # 左上极点,4邻域 # 中心点旁边3个点 sum = int(cur_pixel) \ + int(img[x, y + 1]) \ + int(img[x + 1, y]) \ + int(img[x + 1, y + 1]) if sum <= 2 * 245: img[x, y] = 0 elif x == height - 1: # 右上极点 sum = int(cur_pixel) \ + int(img[x, y + 1]) \ + int(img[x - 1, y]) \ + int(img[x - 1, y + 1]) if sum <= 2 * 245: img[x, y] = 0 else: # 最上非极点,6邻域 sum = int(img[x - 1, y]) \ + int(img[x - 1, y + 1]) \ + int(cur_pixel) \ + int(img[x, y + 1]) \ + int(img[x + 1, y]) \ + int(img[x + 1, y + 1]) if sum <= 3 * 245: img[x, y] = 0 elif y == width - 1: # 最下面一行 if x == 0: # 左下极点 # 中心点旁边3个点 sum = int(cur_pixel) \ + int(img[x + 1, y]) \ + int(img[x + 1, y - 1]) \ + int(img[x, y - 1]) if sum <= 2 * 245: img[x, y] = 0 elif x == height - 1: # 右下极点 sum = int(cur_pixel) \ + int(img[x, y - 1]) \ + int(img[x - 1, y]) \ + int(img[x - 1, y - 1]) if sum <= 2 * 245: img[x, y] = 0 else: # 最下非极点,6邻域 sum = int(cur_pixel) \ + int(img[x - 1, y]) \ + int(img[x + 1, y]) \ + int(img[x, y - 1]) \ + int(img[x - 1, y - 1]) \ + int(img[x + 1, y - 1]) if sum <= 3 * 245: img[x, y] = 0 else: # y不在界线 if x == 0: # 左边非极点 sum = int(img[x, y - 1]) \ + int(cur_pixel) \ + int(img[x, y + 1]) \ + int(img[x + 1, y - 1]) \ + int(img[x + 1, y]) \ + int(img[x + 1, y + 1]) if sum <= 3 * 245: img[x, y] = 0 elif x == height - 1: # 右边非极点 sum = int(img[x, y - 1]) \ + int(cur_pixel) \ + int(img[x, y + 1]) \ + int(img[x - 1, y - 1]) \ + int(img[x - 1, y]) \ + int(img[x - 1, y + 1]) if sum <= 3 * 245: img[x, y] = 0 else: # 具备9规模条件的 sum = int(img[x - 1, y - 1]) \ + int(img[x - 1, y]) \ + int(img[x - 1, y + 1]) \ + int(img[x, y - 1]) \ + int(cur_pixel) \ + int(img[x, y + 1]) \ + int(img[x + 1, y - 1]) \ + int(img[x + 1, y]) \ + int(img[x + 1, y + 1]) if sum <= 4 * 245: img[x, y] = 0 cv2.imwrite(filename,img) return img
结果:
其实到了这一步,这些字符就可以识别了,没须要举办字符切割了,此刻这三种范例的验证码识别率已经到达50%以上了
字符切割
字符切割凡是用于验证码中有粘连的字符,粘连的字符欠好识别,所以我们需要将粘连的字符切割为单个的字符,在举办识别
#p#分页标题#e#
字符切割的思路就是找到一个玄色的点,然后在遍历与他相邻的玄色的点,直到遍历完所有的毗连起来的玄色的点,找出这些点中的最高的点、最低的点、最右边的点、最左边的点,记录下这四个点,认为这是一个字符,然后在向后遍历点,直至找到玄色的点,继承以上的步调。最后通过每个字符的四个点举办切割
图中赤色的点就是代码执行完后,标识出的每个字符的四个点,然后就会按照这四个点举办切割(图中画的有些误差,懂就好)
#p#分页标题#e#
可是也可以看到,m2是粘连的,代码认为他是一个字符,所以我们需要对每个字符的宽度举办检测,假如他的宽渡过宽,我们就认为他是两个粘连在一起的字符,并将它在从中间切割
#p#分页标题#e#
确定每个字符的四个点代码:
def cfs(im,x_fd,y_fd): '''用行列和荟萃记录遍历过的像素坐标取代纯真递归以办理cfs会见过深问题 ''' # print('**********') xaxis=[] yaxis=[] visited =set() q = Queue() q.put((x_fd, y_fd)) visited.add((x_fd, y_fd)) offsets=[(1, 0), (0, 1), (-1, 0), (0, -1)]#四邻域 while not q.empty(): x,y=q.get() for xoffset,yoffset in offsets: x_neighbor,y_neighbor = x+xoffset,y+yoffset if (x_neighbor,y_neighbor) in (visited): continue # 已经会见过了 visited.add((x_neighbor, y_neighbor)) try: if im[x_neighbor, y_neighbor] == 0: xaxis.append(x_neighbor) yaxis.append(y_neighbor) q.put((x_neighbor,y_neighbor)) except IndexError: pass # print(xaxis) if (len(xaxis) == 0 | len(yaxis) == 0): xmax = x_fd + 1 xmin = x_fd ymax = y_fd + 1 ymin = y_fd else: xmax = max(xaxis) xmin = min(xaxis) ymax = max(yaxis) ymin = min(yaxis) #ymin,ymax=sort(yaxis) return ymax,ymin,xmax,xmin def detectFgPix(im,xmax): '''搜索区块起点 ''' h,w = im.shape[:2] for y_fd in range(xmax+1,w): for x_fd in range(h): if im[x_fd,y_fd] == 0: return x_fd,y_fd def CFS(im): '''切割字符位置 ''' zoneL=[]#各区块长度L列表 zoneWB=[]#各区块的X轴[起始,终点]列表 zoneHB=[]#各区块的Y轴[起始,终点]列表 xmax=0#上一区块竣事斑点横坐标,这里是初始化 for i in range(10): try: x_fd,y_fd = detectFgPix(im,xmax) # print(y_fd,x_fd) xmax,xmin,ymax,ymin=cfs(im,x_fd,y_fd) L = xmax - xmin H = ymax - ymin zoneL.append(L) zoneWB.append([xmin,xmax]) zoneHB.append([ymin,ymax]) except TypeError: return zoneL,zoneWB,zoneHB return zoneL,zoneWB,zoneHB
支解粘连字符代码:
# 切割的位置 im_position = CFS(im) maxL = max(im_position[0]) minL = min(im_position[0]) # 假如有粘连字符,假如一个字符的长渡过长就认为是粘连字符,并从中间举办切割 if(maxL > minL + minL * 0.7): maxL_index = im_position[0].index(maxL) minL_index = im_position[0].index(minL) # 配置字符的宽度 im_position[0][maxL_index] = maxL // 2 im_position[0].insert(maxL_index + 1, maxL // 2) # 配置字符X轴[起始,终点]位置 im_position[1][maxL_index][1] = im_position[1][maxL_index][0] + maxL // 2 im_position[1].insert(maxL_index + 1, [im_position[1][maxL_index][1] + 1, im_position[1][maxL_index][1] + 1 + maxL // 2]) # 配置字符的Y轴[起始,终点]位置 im_position[2].insert(maxL_index + 1, im_position[2][maxL_index]) # 切割字符,要想切得好就得设置参数,凡是 1 or 2 就可以 cutting_img(im,im_position,img_name,1,1)
#p#分页标题#e#
切割粘连字符代码:
def cutting_img(im,im_position,img,xoffset = 1,yoffset = 1): filename = './out_img/' + img.split('.')[0] # 识别出的字符个数 im_number = len(im_position[1]) # 切割字符 for i in range(im_number): im_start_X = im_position[1][i][0] - xoffset im_end_X = im_position[1][i][1] + xoffset im_start_Y = im_position[2][i][0] - yoffset im_end_Y = im_position[2][i][1] + yoffset cropped = im[im_start_Y:im_end_Y, im_start_X:im_end_X] cv2.imwrite(filename + '-cutting-' + str(i) + '.jpg',cropped)
结果:
识别
识别用的是typesseract库,主要识别一行字符和单个字符时的参数配置,识别中英文的参数配置,代码很简朴就一行,我这里大多是filter文件的操纵
代码:
# 识别验证码 cutting_img_num = 0 for file in os.listdir('./out_img'): str_img = '' if fnmatch(file, '%s-cutting-*.jpg' % img_name.split('.')[0]): cutting_img_num += 1 for i in range(cutting_img_num): try: file = './out_img/%s-cutting-%s.jpg' % (img_name.split('.')[0], i) # 识别字符 str_img = str_img + image_to_string(Image.open(file),lang = 'eng', config='-psm 10') #单个字符是10,一行文本是7 except Exception as err: pass print('切图:%s' % cutting_img_num) print('识别为:%s' % str_img)
最后这种粘连字符的识别率是在30%阁下,并且这种只是处理惩罚两个字符粘连,假如有两个以上的字符粘连还不能识别,可是按照字符宽度判此外话也不难,有乐趣的可以试一下
无需切割字符识此外结果:
需要切割字符的识别结果:
利用要领:
1、将要识此外验证码图片放入与剧本同级的img文件夹中,建设out_img文件夹
2、python filename
3、二值化、降噪等各个阶段的图片将存储在out_img文件夹中,最终识别功效会打印到屏幕上