https://github.com/python3webspider/deeplearningimagecaptcha
DeepLearning Image Captcha
https://github.com/python3webspider/deeplearningimagecaptcha
Last synced: 2 months ago
JSON representation
DeepLearning Image Captcha
- Host: GitHub
- URL: https://github.com/python3webspider/deeplearningimagecaptcha
- Owner: Python3WebSpider
- License: apache-2.0
- Created: 2021-04-17T17:33:16.000Z (about 4 years ago)
- Default Branch: master
- Last Pushed: 2021-05-04T08:07:14.000Z (about 4 years ago)
- Last Synced: 2025-03-26T17:11:33.823Z (3 months ago)
- Language: Python
- Size: 656 KB
- Stars: 27
- Watchers: 2
- Forks: 24
- Open Issues: 4
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
深度学习识别验证码
=========
Origin Repo: [https://github.com/dee1024/pytorch-captcha-recognition](https://github.com/dee1024/pytorch-captcha-recognition).
本项目致力于使用神经网络来识别各种验证码。
特性
===
- __端到端，不需要做更多的图片预处理（比如图片字符切割、图片尺寸归一化、图片字符标记、字符图片特征提取）__
- __验证码包括数字、大写字母、小写__
- __采用自己生成的验证码来作为神经网络的训练集合、测试集合、预测集合__
- __纯四位数字，验证码识别率高达 99.9999 %__
- __四位数字 + 大写字符，验证码识别率约 96 %__
- __深度学习框架pytorch + 验证码生成器ImageCaptcha__
原理
===
- __训练集合生成__
使用常用的 Python 验证码生成库 ImageCaptcha，生成 10w 个验证码，并且都自动标记好;
如果需要识别其他的验证码也同样的道理，寻找对应的验证码生成算法自动生成已经标记好的训练集合或者手动对标记，需要上万级别的数量，纯手工需要一定的时间，再或者可以借助一些网络的打码平台进行标记
- __训练卷积神经网络__
构建一个多层的卷积网络，进行多标签分类模型的训练
标记的每个字符都做 one-hot 编码
批量输入图片集合和标记数据，大概15个Epoch后，准确率已经达到 96% 以上
验证码识别率展示
========
快速开始
====
- __步骤一：10分钟环境安装__
Python3.5+ 、ImageCaptcha库(pip install captcha)、 Pytorch(参考官网http://pytorch.org)
- __步骤二：生成验证码__
```bash
python generate.py
```
执行以上命令，会在目录 dataset/train/ 下生成多张验证码图片，图片已经标注好，数量可以是 1w、5w、10w，通过 captcha-gen.py 内的 count 参数设定
- __步骤三：训练模型__
```bash
python train.py
```
使用步骤一生成的验证码图集合用CNN模型（在 catcha_cnn_model 中定义）进行训练，训练完成会生成文件 model.pkl
- __步骤四：测试模型__
```bash
python test.py
```
可以在控制台，看到模型的准确率（如 95%） ，如果准确率较低，回到步骤一，生成更多的图片集合再次训练
- __步骤五：使用模型做预测__
```bash
python predict.py
```
可以在控制台，看到预测输出的结果
作者
===
* __Dee Qiu__
* __Germey__