基于python语言的tensorflow的‘端到端’的字符型验证码识别源码整理(github源码分享)

基于python语言的tensorflow的‘端到端’的字符型验证码识别

1   Abstract

验证码（CAPTCHA）的诞生自己是为了自动区分 天然人 和 机器人 的一套公开方法， 可是近几年的人工智能技术的发展，传统的字符验证已经形同虚设。 因此，你们一方面研究和学习此代码时，另一方面也要警戒本身的互联网系统的web安全问题。

Keywords: 人工智能,Python,字符验证码,CAPTCHA,识别,tensorflow,CNN,深度学习

2   Introduction

全自动区分计算机和人类的公开图灵测试（英语：Completely Automated Public Turing test to tell Computers and Humans Apart，簡稱CAPTCHA），俗称验证码，是一种区分用户是计算机或人的公共全自动程序 [1]。

得益于基于卷积神经网络CNN的人工智能技术的发展，目前基于主流的深度学习框架的直接开发出 端到端不分割 的识别方式，并且在没有通过太多trick的状况下，已经能够达到95%以上的识别率。

传统的机器学习方法，对于多位字符验证码都是采用的 化整为零 的方法：先分割成最小单位，再分别识别，而后再统一。 卷积神经网络方法，直接采用 端到端不分割 的方法：输入整张图片，输出整个图片的标记结果，具备更强的通用性。

具体的区别以下图：

端到端 的识别方法显然更具有优点，由于目前的字符型验证码为了防止被识别，多位字符已经彻底融合粘贴在一块儿了，利用传统的技术基本很难实现分割了。本文重点推荐的就是 端到端 的方法。

3   引用声明

本文代码都参考自此文：

http://blog.topspeedsnail.com/archives/10858

斗大的熊猫--《WTF Daily Blog》

本项目主要解决的问题是对某一模式的字符型验证进行端到端的识别。

输入内容：

模型预测结果：

4   本文工做

• 解释了原做者代码注释中提到的关于sigmoid选型的困惑问题并应用到代码中
• 将原做者的代码进行模块工程化，成为总体项目，方便研究的同窗直接进行模式套用

原做者代码中：

def train_crack_captcha_cnn():
    output = crack_captcha_cnn()
    # loss
    #loss = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(output, Y))
    loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=output, labels=Y))
        # 最后一层用来分类的softmax和sigmoid有什么不一样？
    # optimizer 为了加快训练 learning_rate应该开始大，而后慢慢衰
    optimizer = tf.train.AdamOptimizer(learning_rate=0.001).minimize(loss)
    ……

做者在代码的注释中出提出了这样的疑问:

对 softmax 和 sigmoid 的使用方式有疑问。

而后在文章下面读者评论区也都提到了此问题，在此进行总体解释一下。

原文中CNN的输出的维度是 MAX_CAPTCHA*CHAR_SET_LEN ，其实这些维度并不都是彻底独立分布的， 可是使用sigmoid loss也是仍然能够的，至关于先用sigmoid进行了一次归一化，而后再将各个维度的值向目标值进行 回归 ， 最后loss越小，两个向量的对应的值也越接近。 其实sigmoid是能够当作是一个多分类的问题，在这个例子上也能起到比较好的收敛效果。

固然，关于分类的问题，看全部的机器学习框架里面，都是建议使用softmax进行最后的归一化操做，这其实至关因而一种 马太效应 : 让可能性大的分类的值变得更大，让可能性小的份量值变得更小。可是这有个前提，就是参与到softmax运算的一组数据，必须是 相关联 的， 因此若是要使用 softmax_cross_entropy_with_logits ，只须要将网络进行简单修改便可。把输出的维度作成二维[MAX_CAPTCHA, CHAR_SET_LEN], 而后使用softmax loss。

output = crack_captcha_cnn()#36×4
predict = tf.reshape(output, [-1, MAX_CAPTCHA, CHAR_SET_LEN])  # 36行，4列
label = tf.reshape(Y, [-1, MAX_CAPTCHA, CHAR_SET_LEN])

最后使用GPU训练的实验结果对比：

• sigmoid方式。训练6000个step就能达到95%的准确率。
• softmax方式。训练8千个step，达到90%的准确率;训练8万个step，达到94.7%（跑了大半天）

使用tensorboard对accuracy进行监控：

sigmoid-6千个step:

softmax-8千个step:

softmax-8万个step:

总体来讲，在这个例子里面，好像 sigmoid的收敛速度快些，固然这个多是本项目里面的外界因素有利于sigmoid吧，至于具体缘由，等后续再进行研究和解释吧，固然有可能根本解释不了，由于对于CNN，目前主流的意见都是：，反正效果就是好，可是不知道为啥， 科幻得近于玄幻 的一种技术。

github源码地址：

https://github.com/zhengwh/captcha-tensorflow

项目文件介绍：

• cfg.py 配置信息文件
• cnn_sys.py CNN网络结构
• data_iter.py 可迭代的数据集
• gen_captcha.py 验证码生成器，直接使用程序生成带标记的数据
• predict.py 加载训练好的模型，而后对输入的图片进行预测
• train.py 对模型进行训练
• utils.py 一些公共使用的方法

5   小结

本文主要只写原做者没有提到的内容，想了解原文的，能够直接去原做者页面。

6   Reference

[1]	wiki-CAPTCHA https://en.wikipedia.org/wiki/CAPTCHA

7   后续交流

若是有对相关技术有持续关注的兴趣的同窗，欢迎加入QQ群： 592109504

或者手机QQ扫码加入：