CNN识别学库bao的验证码

部署运行你感兴趣的模型镜像

前言

  • 背景
    最近在看深度学习相关知识,正好手上一个爬虫外包有个需求:爬取 学库bao的全站数据。官方传言题目有一千多万道,其中每道题要查看答案和解析,都需要识别验证码(数字+字母,四个字符)。一般的,打码平台一块钱可以识别200张,算下来得要5万的打码费用,还没算失败率。

  • 图片示例
    验证码示例1      验证码示例2    验证码示例3    验证码示例4

  • 参考
    本文参考了《CNN破解简单验证码(Tensorflow实现)》。网上一搜”验证码 CNN”,大部分都是类似的文章,用captcha模块生成验证码进行训练、识别,甚至代码都是一样的,真没意思。后来发现51CTO视频课程里也有讲这个案例,我把它上传到云盘了(链接: https://pan.baidu.com/s/1iuMfSlWyTm8SbxHYeDoBSQ 密码: shjm),感觉视频里的讲解比其他博客详细多了。

  • 代码
    GitHub地址:cnn_on_captchacaptcha_identify

  • 正确率
    训练样本6500,测试样本500,识别率达92.4%。

以往的,识别验证码的流程一般是先去噪,再切割,然后用最近邻去分类识别。但是去噪和切割这两个步骤,经常都不那么好做。而CNN不需要去噪和切割,直接整张验证码拿去训练和识别。不用做针对性的去噪和切割,可以省去很大工作量。



正文

采集样本

CNN识别验证码有优点也有不足,它其中的一个缺陷就是需要更多的样本做训练。文章《识别扭曲干扰性验证码》中提到94%的识别率需要400W张图片,不知真假,但确实吓人。
验证码识别率
难怪网上一搜”CNN 验证码识别”,出来的全都是用captcha模块生成的验证码做训练,敢情他们是找不到训练样本呀。

然而我的任务是识别学库bao的验证码,就算不能收集这么多的验证码,也要试一试。采集验证码的流程是下载验证码 —> 打码平台打码 —> 返回给学库bao判别正误。有些博客说人工识别并标注验证码,有些原始,我用的是云打码,采集了7000张验证码,费用还不到40元,何苦为难自己。另外,打码平台识别的结果并不一定百分百正确,必须返回给原网站,根据反馈判断识别正误。

采集验证码图片的代码:crawl_captcha.py,如果要采集其他网站的验证码,稍加修改即可。
(图片可以自行跑上面的代码采集,我将我采集到的放在云盘了,可以拿来直接用,链接: https://pan.baidu.com/s/18i2xoGth8INChhCWfROZpw 密码: vrew 。如果百度网盘失效就用新浪微盘:http://vdisk.weibo.com/lc/3AO6EQDinZjxwfeVbuZ 密码:L70M)


训练和测试

模型的训练和测试,这个没什么好说的,优快云上一搜,千篇一律。我这里只是将图片源换成了本地采集好的验证码,代码中调一下图片大小。


尝试和改进

嗯,好玩的在这里。无知的初学者,瞎捣鼓系列。
(注意,要先把训练样本和测试样本(captchas文件夹和test文件夹)放到项目目录。除了captcha01,其他都要先运行crop_captcha.py将图片切割成单字符,再进行训练。)

  1. 初版本
    代码:captcha01
    最初,6500张图片,不做任何处理,每次随机抽取64张,训练2W批次。耗时80分钟,准确率3.4%,低的可怜,不忍直视。

  2. 切割
    代码:captcha02
    看来这点数据量,基本上是没法直接训练了。网络模型还没能学习到验证码的特征,那我们就清除一些无用的杂质和边缘,将更清楚、更精准的样本喂给它。去噪我是不想做的,但切割还是比较容易的,把四个字符单独切割出来,把不必要的图片边缘清除掉。这样可以很大程度地降低模型学习的压力,而且样本量也变成原来的四倍了。
    a2    b1    c    d    e    f    g     h    i    j    k
    切割以后总共2.6万张图,训练2W批次,耗时23分钟,准确率82.2%(四个字符同时正确)。这准确率立马上来了,基本上可以拿来用了。
    (注:因打码平台经常将大小写忽略,故训练的时候将全部大写字母作小写看待。)

  3. 除去pooling层
    代码:captcha03
    切割后的图片大小为6*18,字符已经精细到1个像素上去了,已经不能再缩减了。而pooling层很重要的一个作用就是在保留主要特征的同时降低参数(纬度)和计算量,此时这个作用并不大,所以可以考虑把pooling层去掉。
    这里写图片描述     这里写图片描述    这里写图片描述    这里写图片描述
    训练2W批次,耗时37分钟,准确率89.2%

  4. 新增一层网络
    代码:captcha04
    验证码的识别,一般的选择三层卷积就已经有比较好的效果,但我还是好奇四层卷积,识别率将会升降多少。
    训练2W批次,耗时44分钟,准确率91.2%
    训练6W批次,耗时135分钟,准确率91.6%

  5. 将第三层的厚度64变成128
    代码:captcha05
    训练2W批次,耗时53分钟,准确率88.8%
    训练6W批次,耗时207分钟,准确率92.4%


开启验证码识别服务

好了,模型已经训练好了。接下来就是使用了。我使用了Django开了一个服务,需要识别验证码的时候只需将验证码post过去就能返回识别结果了。代码:captcha_identify
启动命令:python manage.py runserver 127.0.0.1:8000

在爬虫中,只需要调用下面的方法即可。

def identify(filename):
    try:
        r = requests.post('http://127.0.0.1:8000/captcha_identify/', files={'image': (filename, open(filename, 'rb'), 'image/png')})
        return r.content
    except Exception as e:
        return ''



结语

92.4%的识别率,这已经算比较高了,毕竟训练样本才这么些。而且识别率跟噪声有关,这个网站有些验证码的噪声线条,已经让字符完全无法识别了。所以识别率提高到一定程度就很难再提高了。
无法识别1    无法识别2    无法识别3    无法识别4

一年多没写文章了,有点生疏。本文借鉴captcha验证码的识别案例,记录识别学库bao验证码的过程,目的是总结和分享,学艺不精,纯属捣鼓,错误之处各位多批评指出。

您可能感兴趣的与本文相关的镜像

Yolo-v5

Yolo-v5

Yolo

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎

点选验证码训练是为了让模型能够准确识别并按照要求完成点选操作,以下是一些常见的训练方法和相关技术: ### 训练方法 #### 监督习 在监督习方法中,需要准备大量带有标注的训练数据。这些数据包含验证码图片以及对应的正确点选位置或目标信息。例如,对于要求点选特定物体的验证码,标注会明确指出图片中该物体的位置。使用卷积神经网络(CNN)作为基础模型,将标注好的图片输入到网络中进行训练。通过训练,模型习到图片特征与正确点选结果之间的映射关系。训练过程中会多次迭代,不断调整模型的参数,以最小化预测结果与标注结果之间的误差。常用的损失函数如交叉熵损失函数可以衡量模型预测的准确性,通过反向传播算法更新模型参数,使损失函数值逐渐降低,从而提高模型的性能。 ```python import torch import torch.nn as nn import torch.optim as optim from torchvision import datasets, transforms from torch.utils.data import DataLoader # 定义简单的CNN模型 class SimpleCNN(nn.Module): def __init__(self): super(SimpleCNN, self).__init__() self.conv1 = nn.Conv2d(3, 16, kernel_size=3, padding=1) self.relu1 = nn.ReLU() self.pool1 = nn.MaxPool2d(2) self.fc1 = nn.Linear(16 * 16 * 16, 10) def forward(self, x): x = self.pool1(self.relu1(self.conv1(x))) x = x.view(-1, 16 * 16 * 16) x = self.fc1(x) return x # 数据预处理 transform = transforms.Compose([ transforms.Resize((32, 32)), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) # 加载数据集(假设已经有标注好的数据集) train_dataset = datasets.ImageFolder(root='path/to/train_data', transform=transform) train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True) # 初始化模型、损失函数和优化器 model = SimpleCNN() criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9) # 训练模型 for epoch in range(10): running_loss = 0.0 for i, data in enumerate(train_loader, 0): inputs, labels = data optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print(f'Epoch {epoch + 1}, Loss: {running_loss / len(train_loader)}') ``` #### 迁移习 迁移习是利用在大规模数据集上预训练好的模型,将其应用到点选验证码训练任务中。由于预训练模型已经习到了丰富的图像特征表示,在点选验证码训练时,可以在预训练模型的基础上进行微调。保留预训练模型的大部分结构和参数,只对最后几层进行修改和训练。这样可以减少训练所需的数据量和时间,同时提高模型的泛化能力。例如,使用在ImageNet数据集上预训练好的ResNet模型,将其最后一层全连接层替换为适用于点选验证码任务的输出层,然后在点选验证码数据集上进行微调训练。 ```python import torch import torch.nn as nn import torch.optim as optim from torchvision import models, datasets, transforms from torch.utils.data import DataLoader # 加载预训练的ResNet模型 model = models.resnet18(pretrained=True) num_ftrs = model.fc.in_features model.fc = nn.Linear(num_ftrs, 10) # 假设是10分类的点选验证码任务 # 数据预处理 transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) # 加载数据集(假设已经有标注好的数据集) train_dataset = datasets.ImageFolder(root='path/to/train_data', transform=transform) train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True) # 定义损失函数和优化器 criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9) # 训练模型 for epoch in range(10): running_loss = 0.0 for i, data in enumerate(train_loader, 0): inputs, labels = data optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print(f'Epoch {epoch + 1}, Loss: {running_loss / len(train_loader)}') ``` ### 相关技术 #### 图像增强技术 为了增加训练数据的多样性,提高模型的泛化能力,会使用图像增强技术对训练数据进行处理。常见的图像增强方法包括旋转、翻转、缩放、亮度调整、对比度调整等。例如,将验证码图片随机旋转一定角度、水平或垂直翻转、调整亮度和对比度等操作,生成新的训练样本。这样可以模拟不同环境下的验证码情况,让模型习到更鲁棒的特征。 ```python import torchvision.transforms as transforms # 定义图像增强变换 transform = transforms.Compose([ transforms.RandomRotation(15), transforms.RandomHorizontalFlip(), transforms.ColorJitter(brightness=0.2, contrast=0.2, saturation=0.2, hue=0.1), transforms.ToTensor() ]) ``` #### 目标检测技术 对于点选验证码中需要识别特定目标的任务,目标检测技术可以发挥重要作用。目标检测算法如Faster R - CNN、YOLO等可以在图片中定位出目标物体的位置。在训练过程中,使用这些目标检测算法对验证码图片进行处理,得到目标物体的边界框信息。将这些信息作为辅助特征输入到点选验证码模型中,帮助模型更准确地进行点选判断。 #### 注意力机制 注意力机制可以让模型更加关注图片中与点选任务相关的区域。在模型中引入注意力模块,如SE(Squeeze - and - Excitation)模块、CBAM(Convolutional Block Attention Module)等。这些模块可以自适应地调整特征图的通道和空间注意力权重,使模型更加聚焦于重要的特征区域,提高点选的准确性。
评论 9
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值