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