python+selenium+PIL+tesseract验证码识别

1 from selenium import webdriver
2 from time import sleep
3 import unittest
4 from PIL import Image
5 from PIL import ImageEnhance
6 import pytesseract
7 driver&＃61;webdriver.Firefox()
8 url&＃61;"https://passport.baidu.com/?getpassindex"
9 driver.get(url)
10 driver.maximize_window()
11 driver.save_screenshot(r"E:\aa.png") #截取当前网页&＃xff0c;该网页有我们需要的验证码
12 imgelement &＃61; driver.find_element_by_xpath(".//*[&＃64;id&＃61;&＃39;forgotsel&＃39;]/div/div[3]/img")
13 #imgelement &＃61; driver.find_element_by_id("code") #定位验证码
14 location &＃61; imgelement.location #获取验证码x,y轴坐标
15 print (location)
16 size&＃61;imgelement.size #获取验证码的长宽
17 print(size)
18 coderange&＃61;(int(location[&＃39;x&＃39;]),int(location[&＃39;y&＃39;]),int(location[&＃39;x&＃39;]&＃43;size[&＃39;width&＃39;]),
19 int(location[&＃39;y&＃39;]&＃43;size[&＃39;height&＃39;])) #写成我们需要截取的位置坐标
20 i&＃61;Image.open(r"E:\aa.png") #打开截图
21 frame4&＃61;i.crop(coderange) #使用Image的crop函数&＃xff0c;从截图中再次截取我们需要的区域
22 frame4.save(r"E:\frame4.png")
23 i2&＃61;Image.open(r"E:\frame4.png")
24 imgry &＃61; i2.convert(&＃39;L&＃39;) #图像加强&＃xff0c;二值化&＃xff0c;PIL中有九种不同模式。分别为1&＃xff0c;L&＃xff0c;P&＃xff0c;RGB&＃xff0c;RGBA&＃xff0c;CMYK&＃xff0c;YCbCr&＃xff0c;I&＃xff0c;F。L为灰度图像
25 sharpness &＃61;ImageEnhance.Contrast(imgry)#对比度增强
26 i3 &＃61; sharpness.enhance(3.0) #3.0为图像的饱和度
27 i3.save("E:\\image_code.png")
28 i4&＃61;Image.open("E:\\image_code.png")
29 text&＃61;pytesseract.image_to_string(i4)#使用image_to_string识别验证码
30 print (text)