溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

使用python PIL庫實現(xiàn)簡單驗證碼的去噪方法步驟

發(fā)布時間:2020-09-18 19:19:16 來源:腳本之家 閱讀:168 作者:鍋爐房劉大爺 欄目:開發(fā)技術(shù)

字符型圖片驗證碼識別完整過程及Python實現(xiàn)的博主,我的大部分知識點都是從他那里學(xué)來的。

想要識別驗證碼,收集足夠多的樣本后,首先要做的就是對驗證碼原始圖片進行處理,對驗證碼識別分類之前,一般包括:將彩色圖片轉(zhuǎn)換成灰度圖、將灰度圖二值化和去除噪點三個基本過程。這里僅以比較簡單的驗證碼為例,介紹一下如何通過python的PIL庫對圖片去噪。

首先看一下未經(jīng)處理的驗證碼圖片:

使用python PIL庫實現(xiàn)簡單驗證碼的去噪方法步驟

對圖片處理主要使用了PIL庫的Image類。

1.彩色圖片轉(zhuǎn)換成灰度圖

首先使用Image的open方法打開上面的圖片,可以得到一個PIL.Image.Image對象,之后就可以調(diào)用convert、filter、point和putpixel等方法來處理圖片。

我們可以通過convert方法將上面的彩色圖片轉(zhuǎn)換成灰度圖:

# encoding=utf8
 
from PIL import Image
 
def main():
	image = Image.open('RandomPicture.png')
	imgry = image.convert('L')
	imgry.save('gray.png')
 
if __name__ == '__main__':
	main()

運行結(jié)果:

使用python PIL庫實現(xiàn)簡單驗證碼的去噪方法步驟

通過保存的圖片可以看出來,已經(jīng)由原來的彩圖變成了灰度圖,或者也可以認(rèn)為是黑白圖。什么叫灰度圖?我們知道彩色圖片是由不同的顏色的像素組合到一起的,那灰度圖可以類似的認(rèn)為是由不同灰度值的像素組合在一起后呈現(xiàn)出來的。任何顏色都有紅、綠、藍三原色組成,假如原來某點的顏色為RGB(R,G,B),那么,我們可以通過下面幾種方法,將其轉(zhuǎn)換為灰度:

1.浮點算法:Gray=R*0.3+G*0.59+B*0.11
2.整數(shù)方法:Gray=(R*30+G*59+B*11)/100
3.移位方法:Gray =(R*76+G*151+B*28)>>8;
4.平均值法:Gray=(R+G+B)/3;
5.僅取綠色:Gray=G;

通過上述任一種方法求得Gray后,將原來的RGB(R,G,B)中的R,G,B統(tǒng)一用Gray替換,形成新的顏色RGB(Gray,Gray,Gray),用它替換原來的RGB(R,G,B)就是灰度圖了。

用代碼實現(xiàn)看一下:

# encoding=utf8
 
from PIL import Image
 
def main():
	image = Image.open('RandomPicture.png')
	print 'image mode: ', image.mode
	print image.getpixel((0, 0))
	print '-' * 40
	imgry = image.convert('L')
	print 'imgry mode: ', imgry.mode
	print imgry.getpixel((0, 0))
 
if __name__ == '__main__':
	main()

運行結(jié)果:

image mode: RGB
(21, 10, 26)
----------------------------------------
imgry mode: L
15

代碼說明:

通過image.mode方法可以獲得當(dāng)前的PIL.Image.Image對象(也就是當(dāng)前打開的圖片)的mode值,而mode值表示圖片的單位顏色是由RGB三個值組成的還是由灰度值組成的;

而getpixel可以獲取某個像素的RGB值或者灰度值。我們知道圖片是由許多像素組成的,每個像素在圖片上都有一個對應(yīng)的坐標(biāo)x和y,而“(0, 0)”就表示該圖片左上角頂點的像素。

由上面的結(jié)果我們可以知道,在將圖片轉(zhuǎn)換成灰度圖之前,“(0, 0)”代表的像素的顏色是由RGB組成的:(21, 10, 26);在通過concert將彩色圖片轉(zhuǎn)換成灰度后,“(0, 0)”代表的像素的顏色值就變成了一個值:“15”,通過打印imgry.mode我們也可以知道,此時圖片已經(jīng)變成了灰度圖,它的每一個像素的顏色都變成了一個灰度值。

其實這時候我們也可以簡單的計算一下,使用前面說的浮點算法將上面的(21, 10, 26)三個值帶入計算:

>>> 21*0.3+10*0.59+26*0.11
15.059999999999999

結(jié)果顯示確實由浮點算法將RGB值變成了灰度值。

2.灰度圖二值化

我們已經(jīng)得到了灰度圖,接下來就是將灰度圖二值化。所謂二值化就是將灰度圖像轉(zhuǎn)換成由黑白二色組成的圖像。思路就是確定一個閾值,大于閾值的像素表示為白色,小于閾值的像素表示為黑色,以此將圖片的像素(灰度值)劃分為兩部分:0和1,例如0代表黑色,1代表白色,然后我們就可以用一串0和1組成的數(shù)字來表示一張圖片。

將灰度圖二值化會用到point方法,它可以接收一個灰度轉(zhuǎn)二值的映射table,具體原理暫時還沒弄明白,代碼實現(xiàn)過程是這樣的:

# encoding=utf8
 
from PIL import Image
 
def get_bin_table(threshold=115):
	'''
	獲取灰度轉(zhuǎn)二值的映射table
	0表示黑色,1表示白色
	'''
	table = []
	for i in range(256):
		if i < threshold:
			table.append(0)
		else:
			table.append(1)
	return table
 
def main():
	image = Image.open('RandomPicture.png')
	imgry = image.convert('L')
	table = get_bin_table()
	binary = imgry.point(table, '1')
	binary.save('binary.png')
 
if __name__ == '__main__':
	main()

運行結(jié)果:

使用python PIL庫實現(xiàn)簡單驗證碼的去噪方法步驟

通過結(jié)果不難看出,我們已經(jīng)將最開始的彩色圖由灰度圖轉(zhuǎn)變成了僅由黑白二色組成的圖片,實現(xiàn)了二值化。這里需要說明的是,threshold參數(shù)值針對當(dāng)前的驗證碼圖片合適,該值需要根據(jù)驗證碼類型不同進行調(diào)試來確定。

然后我們再看看(0, 0)坐標(biāo)代表的像素的顏色值是什么:

# encoding=utf8
 
from PIL import Image
 
def get_bin_table(threshold=115):
	'''
	獲取灰度轉(zhuǎn)二值的映射table
	0表示黑色,1表示白色
	'''
	table = []
	for i in range(256):
		if i < threshold:
			table.append(0)
		else:
			table.append(1)
	return table
 
def main():
	image = Image.open('RandomPicture.png')
	print 'image mode: ', image.mode
	print image.getpixel((0, 0))
	co = image.getcolors()
	print co
	print '-' * 40
	imgry = image.convert('L')
	print 'imgry mode: ', imgry.mode
	print imgry.getpixel((0, 0))
	co = imgry.getcolors()
	print co
	print '-' * 40
	table = get_bin_table()
	binary = imgry.point(table, '1')
	print 'binary mode: ', binary.mode
	print binary.getpixel((0, 0))
	co = binary.getcolors()
	print co
 
if __name__ == '__main__':
	main()

運行結(jié)果:

image mode: RGB
(21, 10, 26)
None
----------------------------------------
imgry mode: L
15
[(1, 2), (2, 3), (1, 4), (1, 5), (4, 6), (3, 8), (3, 9), (6, 10), (4, 11), (4, 12), (7, 13), (8, 14), (3, 15), (12, 16), (7, 17), (6, 18), (5, 19), (13, 20), (9, 21), (9, 22), (4, 23), (5, 24), (7, 25), (3, 26), (6, 27), (7, 28), (3, 29), (3, 30), (3, 31), (5, 32), (1, 33), (3, 35), (2, 36), (2, 37), (2, 38), (2, 39), (1, 41), (3, 42), (1, 43), (2, 44), (7, 45), (3, 46), (5, 47), (1, 48), (3, 49), (3, 50), (3, 51), (5, 52), (4, 53), (1, 54), (7, 55), (7, 56), (10, 57), (4, 58), (5, 59), (6, 60), (5, 61), (12, 62), (7, 63), (10, 64), (12, 65), (14, 66), (15, 67), (11, 68), (9, 69), (11, 70), (7, 71), (9, 72), (5, 73), (10, 74), (5, 75), (5, 76), (5, 77), (8, 78), (7, 79), (3, 80), (5, 81), (6, 82), (5, 83), (3, 84), (3, 85), (6, 86), (2, 87), (3, 88), (2, 90), (3, 91), (1, 93), (2, 94), (3, 95), (1, 96), (3, 97), (2, 99), (3, 100), (3, 101), (1, 102), (3, 104), (4, 105), (1, 106), (3, 108), (4, 110), (4, 111), (4, 112), (3, 113), (3, 114), (5, 115), (2, 116), (3, 117), (8, 118), (8, 119), (8, 120), (7, 121), (9, 122), (9, 123), (11, 124), (11, 125), (2, 126), (10, 127), (9, 128), (7, 129), (13, 130), (11, 131), (11, 132), (9, 133), (16, 134), (11, 135), (12, 136), (8, 137), (14, 138), (12, 139), (13, 140), (20, 141), (22, 142), (19, 143), (14, 144), (23, 145), (17, 146), (10, 147), (18, 148), (13, 149), (11, 150), (26, 151), (16, 152), (14, 153), (11, 154), (17, 155), (10, 156), (12, 157), (12, 158), (20, 159), (18, 160), (16, 161), (22, 162), (20, 163), (16, 164), (13, 165), (14, 166), (13, 167), (11, 168), (17, 169), (8, 170), (16, 171), (20, 172), (12, 173), (10, 174), (10, 175), (10, 176), (11, 177), (7, 178), (8, 179), (7, 180), (5, 181), (7, 182), (4, 183), (7, 184), (4, 185), (4, 186), (5, 187), (6, 188), (2, 189), (1, 190), (4, 191), (6, 192), (12, 193), (8, 194), (10, 195), (3, 196), (13, 197), (9, 198), (19, 199), (18, 200), (20, 201), (16, 202), (18, 203), (24, 204), (33, 205), (25, 206), (33, 207), (38, 208), (31, 209), (46, 210), (39, 211), (53, 212), (54, 213), (33, 214), (42, 215), (54, 216), (60, 217), (50, 218), (36, 219), (48, 220), (32, 221), (45, 222), (28, 223), (24, 224), (21, 225), (19, 226), (21, 227), (13, 228), (12, 229), (12, 230), (13, 231), (5, 232), (8, 233), (4, 234), (5, 235), (1, 236), (1, 237), (2, 238), (1, 239), (1, 240), (1, 242), (1, 243)]
----------------------------------------
binary mode: 1
0
[(503, 0), (1993, 1)]

代碼說明:

通過binary mode的值我們可以知道二值化后得到的圖片的像素值由0或1表示,而且當(dāng)前(0, 0)代表的像素值的為0,它代表黑色,通過上面的圖片我們也可以知道,左上角頂點確實是黑色的。

上面的代碼中我們還使用了getcolors方法,它用來返回像素信息,是一個含有元素的列表:[(該種像素的數(shù)量,(該種像素)),(...),...],當(dāng)該列表特別大的時候,它會返回None,這也是為什么上面彩色圖片調(diào)用getcolors的時候會返回None。而[(503, 0), (1993, 1)]就表示我們得到的二值化黑白圖片,由503個黑色像素點和1993個白色像素點組成。

通過binary.size我們可以得到二值化后的黑白圖片的width和height值:(78, 32),它就表示該圖片由78X32個像素點組成,正好等于503+1993的和。(78, 32)也說明該圖片橫向上共有32行,每行有78個像素點。將由0和1表示的圖片打印出來看一下:

# encoding=utf8
 
from PIL import Image
 
def get_bin_table(threshold=115):
	'''
	獲取灰度轉(zhuǎn)二值的映射table
	0表示黑色,1表示白色
	'''
	table = []
	for i in range(256):
		if i < threshold:
			table.append(0)
		else:
			table.append(1)
	return table
 
def main():
	image = Image.open('RandomPicture.png')
	imgry = image.convert('L')
	table = get_bin_table()
	binary = imgry.point(table, '1')
	width, height = binary.size
	lis = binary.getdata()	# 返回圖片所有的像素值,要使用list()才能顯示出具體數(shù)值
	lis = list(lis)
	start = 0
	step = width
	for i in range(height):
		for p in lis[start: start+step]:
			if p == 1:	# 將白色的點變成空格,方便人眼看
				p = ' '
			print p,
		print
		start += step
 
if __name__ == '__main__':
	main()

運行結(jié)果:

使用python PIL庫實現(xiàn)簡單驗證碼的去噪方法步驟

通過上面的結(jié)果已經(jīng)大致可以看出該圖片表示的就是“959c”。

3.去除噪點

由上面的結(jié)果也不難看出,除了表示“959c”的“0”,圖片中還有其他的“0”代表的“噪點”,我們要盡可能的去除它們,方便后期的識別訓(xùn)練。

對于去除噪點,我這里也是借鑒了字符型圖片驗證碼識別完整過程及Python實現(xiàn)里面的“九宮格”方法,代碼實現(xiàn):

# encoding=utf8
 
from PIL import Image
 
def sum_9_region_new(img, x, y):
	'''確定噪點 '''
	cur_pixel = img.getpixel((x, y)) # 當(dāng)前像素點的值
	width = img.width
	height = img.height
 
	if cur_pixel == 1: # 如果當(dāng)前點為白色區(qū)域,則不統(tǒng)計鄰域值
		return 0
 
	# 因當(dāng)前圖片的四周都有黑點,所以周圍的黑點可以去除
	if y < 3: # 本例中,前兩行的黑點都可以去除
		return 1
	elif y > height - 3: # 最下面兩行
		return 1
	else: # y不在邊界
		if x < 3: # 前兩列
			return 1
		elif x == width - 1: # 右邊非頂點
			return 1
		else: # 具備9領(lǐng)域條件的
			sum = img.getpixel((x - 1, y - 1)) \
				 + img.getpixel((x - 1, y)) \
				 + img.getpixel((x - 1, y + 1)) \
				 + img.getpixel((x, y - 1)) \
				 + cur_pixel \
				 + img.getpixel((x, y + 1)) \
				 + img.getpixel((x + 1, y - 1)) \
				 + img.getpixel((x + 1, y)) \
				 + img.getpixel((x + 1, y + 1))
			return 9 - sum
 
def collect_noise_point(img):
	'''收集所有的噪點'''
	noise_point_list = []
	for x in range(img.width):
		for y in range(img.height):
			res_9 = sum_9_region_new(img, x, y)
			if (0 < res_9 < 3) and img.getpixel((x, y)) == 0: # 找到孤立點
				pos = (x, y)
				noise_point_list.append(pos)
	return noise_point_list
 
def remove_noise_pixel(img, noise_point_list):
	'''根據(jù)噪點的位置信息,消除二值圖片的黑點噪聲'''
	for item in noise_point_list:
		img.putpixel((item[0], item[1]), 1)
 
def get_bin_table(threshold=115):
	'''獲取灰度轉(zhuǎn)二值的映射table,0表示黑色,1表示白色'''
	table = []
	for i in range(256):
		if i < threshold:
			table.append(0)
		else:
			table.append(1)
	return table
 
def main():
	image = Image.open('RandomPicture.png')
	imgry = image.convert('L')
	table = get_bin_table()
	binary = imgry.point(table, '1')
	noise_point_list = collect_noise_point(binary)
	remove_noise_pixel(binary, noise_point_list)
	binary.save('finaly.png')
 
if __name__ == '__main__':
	main()

運行結(jié)果:

使用python PIL庫實現(xiàn)簡單驗證碼的去噪方法步驟

通過截圖可以知道,我們已經(jīng)去除了圖片四周的噪點以及一些孤立的噪點。

還要再說一句的就是,除了上面的步驟,我們還可以通過PIL的ImageEnhance,和ImageFilter對圖片做其他處理(例如增加對比度、亮度、銳化等),這里就不舉例說明了,因為不同的圖片經(jīng)過這些處理后,可能效果會不同。

最后,要感謝這些博主的分享,為我學(xué)習(xí)驗證碼識別相關(guān)知識提供了不少參考:

字符型圖片驗證碼識別完整過程及Python實現(xiàn)

Python3.5+sklearn 使用SVM自動識別字母驗證碼

python簡單驗證碼識別

基于Python的PIL庫學(xué)習(xí)(一)

以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持億速云。

向AI問一下細節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI