您好,登錄后才能下訂單哦!
字符型圖片驗證碼識別完整過程及Python實現的博主,我的大部分知識點都是從他那里學來的。
想要識別驗證碼,收集足夠多的樣本后,首先要做的就是對驗證碼原始圖片進行處理,對驗證碼識別分類之前,一般包括:將彩色圖片轉換成灰度圖、將灰度圖二值化和去除噪點三個基本過程。這里僅以比較簡單的驗證碼為例,介紹一下如何通過python的PIL庫對圖片去噪。
首先看一下未經處理的驗證碼圖片:
對圖片處理主要使用了PIL庫的Image類。
1.彩色圖片轉換成灰度圖
首先使用Image的open方法打開上面的圖片,可以得到一個PIL.Image.Image對象,之后就可以調用convert、filter、point和putpixel等方法來處理圖片。
我們可以通過convert方法將上面的彩色圖片轉換成灰度圖:
# 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()
運行結果:
通過保存的圖片可以看出來,已經由原來的彩圖變成了灰度圖,或者也可以認為是黑白圖。什么叫灰度圖?我們知道彩色圖片是由不同的顏色的像素組合到一起的,那灰度圖可以類似的認為是由不同灰度值的像素組合在一起后呈現出來的。任何顏色都有紅、綠、藍三原色組成,假如原來某點的顏色為RGB(R,G,B),那么,我們可以通過下面幾種方法,將其轉換為灰度:
1.浮點算法:Gray=R*0.3+G*0.59+B*0.11
2.整數方法: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統一用Gray替換,形成新的顏色RGB(Gray,Gray,Gray),用它替換原來的RGB(R,G,B)就是灰度圖了。
用代碼實現看一下:
# 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()
運行結果:
image mode: RGB (21, 10, 26) ---------------------------------------- imgry mode: L 15
代碼說明:
通過image.mode方法可以獲得當前的PIL.Image.Image對象(也就是當前打開的圖片)的mode值,而mode值表示圖片的單位顏色是由RGB三個值組成的還是由灰度值組成的;
而getpixel可以獲取某個像素的RGB值或者灰度值。我們知道圖片是由許多像素組成的,每個像素在圖片上都有一個對應的坐標x和y,而“(0, 0)”就表示該圖片左上角頂點的像素。
由上面的結果我們可以知道,在將圖片轉換成灰度圖之前,“(0, 0)”代表的像素的顏色是由RGB組成的:(21, 10, 26);在通過concert將彩色圖片轉換成灰度后,“(0, 0)”代表的像素的顏色值就變成了一個值:“15”,通過打印imgry.mode我們也可以知道,此時圖片已經變成了灰度圖,它的每一個像素的顏色都變成了一個灰度值。
其實這時候我們也可以簡單的計算一下,使用前面說的浮點算法將上面的(21, 10, 26)三個值帶入計算:
>>> 21*0.3+10*0.59+26*0.11 15.059999999999999
結果顯示確實由浮點算法將RGB值變成了灰度值。
2.灰度圖二值化
我們已經得到了灰度圖,接下來就是將灰度圖二值化。所謂二值化就是將灰度圖像轉換成由黑白二色組成的圖像。思路就是確定一個閾值,大于閾值的像素表示為白色,小于閾值的像素表示為黑色,以此將圖片的像素(灰度值)劃分為兩部分:0和1,例如0代表黑色,1代表白色,然后我們就可以用一串0和1組成的數字來表示一張圖片。
將灰度圖二值化會用到point方法,它可以接收一個灰度轉二值的映射table,具體原理暫時還沒弄明白,代碼實現過程是這樣的:
# encoding=utf8 from PIL import Image def get_bin_table(threshold=115): ''' 獲取灰度轉二值的映射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()
運行結果:
通過結果不難看出,我們已經將最開始的彩色圖由灰度圖轉變成了僅由黑白二色組成的圖片,實現了二值化。這里需要說明的是,threshold參數值針對當前的驗證碼圖片合適,該值需要根據驗證碼類型不同進行調試來確定。
然后我們再看看(0, 0)坐標代表的像素的顏色值是什么:
# encoding=utf8 from PIL import Image def get_bin_table(threshold=115): ''' 獲取灰度轉二值的映射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()
運行結果:
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表示,而且當前(0, 0)代表的像素值的為0,它代表黑色,通過上面的圖片我們也可以知道,左上角頂點確實是黑色的。
上面的代碼中我們還使用了getcolors方法,它用來返回像素信息,是一個含有元素的列表:[(該種像素的數量,(該種像素)),(...),...],當該列表特別大的時候,它會返回None,這也是為什么上面彩色圖片調用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): ''' 獲取灰度轉二值的映射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()才能顯示出具體數值 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()
運行結果:
通過上面的結果已經大致可以看出該圖片表示的就是“959c”。
3.去除噪點
由上面的結果也不難看出,除了表示“959c”的“0”,圖片中還有其他的“0”代表的“噪點”,我們要盡可能的去除它們,方便后期的識別訓練。
對于去除噪點,我這里也是借鑒了字符型圖片驗證碼識別完整過程及Python實現里面的“九宮格”方法,代碼實現:
# encoding=utf8 from PIL import Image def sum_9_region_new(img, x, y): '''確定噪點 ''' cur_pixel = img.getpixel((x, y)) # 當前像素點的值 width = img.width height = img.height if cur_pixel == 1: # 如果當前點為白色區域,則不統計鄰域值 return 0 # 因當前圖片的四周都有黑點,所以周圍的黑點可以去除 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領域條件的 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): '''根據噪點的位置信息,消除二值圖片的黑點噪聲''' for item in noise_point_list: img.putpixel((item[0], item[1]), 1) def get_bin_table(threshold=115): '''獲取灰度轉二值的映射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()
運行結果:
通過截圖可以知道,我們已經去除了圖片四周的噪點以及一些孤立的噪點。
還要再說一句的就是,除了上面的步驟,我們還可以通過PIL的ImageEnhance,和ImageFilter對圖片做其他處理(例如增加對比度、亮度、銳化等),這里就不舉例說明了,因為不同的圖片經過這些處理后,可能效果會不同。
最后,要感謝這些博主的分享,為我學習驗證碼識別相關知識提供了不少參考:
字符型圖片驗證碼識別完整過程及Python實現
Python3.5+sklearn 使用SVM自動識別字母驗證碼
python簡單驗證碼識別
基于Python的PIL庫學習(一)
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。