前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >银行卡号定位与识别系统

银行卡号定位与识别系统

作者头像
机器学习AI算法工程
发布2021-07-14 11:22:43
1.9K0
发布2021-07-14 11:22:43
举报
功能介绍

在这里因为给定的数据集都是每一个上有四个数字(有些为空格),但是所识别的最终目标——银行卡号有不定的长度。现在比较流行的解决方案是CRNN和CTC损失函数。于是我就这么做吧,CNN部分参考自VGG网络,RNN则使用的是双向LSTM(简称BLSTM),使用GRU也可以实现,可以自己试试。

这个模型不好的地方是训练。训练参数可以在crnn/cfg.py中设置(推荐使用默认设置),性能较为一般的显卡推荐使用默认设置,性能卓越的显卡可以增大BATCH_SIZE,以加快训练速度。识别效果如下图:

模型的训练方法如下:

python crnn/preprocess.py

python crnn/run.py

注1:模型训练的数据集是由子数据集(4-6个)拼凑而成的不定长度的数据,crnn/cfg.py中的MAX_LABEL_LENGTH用来设置数据的最大长度。这部分数据是通过运行crnn/preprocess.py生成,可以通过AUG_NBR设置生成数据集个数。其效果如下:

注2:crnn/preprocess.py直接将可训练数据分开打包到.npz文件中(文件中所容纳最大数据数量可在crnn/cfg.py中的PACK_NBR_MAX设置)。训练数据虽然对我们来说是透明的,但可减少训练时部分重复操作、系统所做I/O,进而增大训练速度和GPU利用率。

注3:模型训练在EPOCH=10以内就已经收敛。最终loss和val_loss值近似在1.73左右。

项目 代码 和 数据集 获取方式:

搜索公众号关注: datanlp

卡号定位

这一部分是重中之重,考虑到自己还没有能力去研究新的图像字符定位算法,就参考了许多这类算法。现在比较成功的莫过于CTPN了,其次也有Faster-RCNN,EAST,Seg-Link等等。而偏偏选用EAST的原因可能是一见钟情吧(名字好听)。

此部分引用的是EAST的Keras实现,但此EAST并非原生的EAST,而是改进的AdvancedEAST

这个模型特点之一就是方便训练,直观也简练。

特别值得提起的是数据集,是我自己从互联网上搜罗来的,然后还进行了标注。过程比较痛苦。面临的问题是,几百张数据可能训练不出好的模型,所以后期使用自动定位不成功的,可能原因之一就是数据集太小(我…尽力了…)。但是用于这样的小系统开发,还是有一定的使用价值。

如果你有更多的数据的话,若想进行标注。懂的同学可以标注直接标注为 ICDAR 2015的格式,不懂的同学则可以使用dataset/tagger.py简陋的标注器来标注,标注方法:以目标区域左上角开始,顺时针的点击物体的角点,直到最后一个点,可在数据同目录的txt下看到形如x1,y1,x2,y2,x3,y3,x4,y4,###的值###表示忽略目标内容。如果成功标注完成,在east目录下运行python east/preprocess.py得到形如以下形式的图像,就代表成功了。

模型的训练方法如下:

  1. python east/preprocess.py(可选,若数据集有变更则运行)
  2. python east/run.py

注1:部分参数可以在east/cfg.py中设置,推荐使用默认。

交互界面

使用了Qt对Python的接口PyQt5来完成,UI设计使用QtDesigner来完成。整体效果如下:(且不说美,起码能看,中规中矩吧)

除了读取(Load)和识别(Identify)两个按钮外,还有复制(Copy)和一些调整视图的按钮,旋转和缩放,基本够用了。 把定位和识别功能都集成在了Identify按钮上,点击Load读取照片后,直接点击这个按钮就好。效果如下:

但是也想之前所述,因为训练数据的贫乏,导致部分测试数据无法定位。后期可通过增大训练数据集解决。目前自动定位失败主要集中在卡号和卡片背景颜色相近的情况,例如都偏暗,或者都偏亮。可以在定位之前,给测试数据加上若干预处理操作,如提高亮度,增加对比度等

如果自动识别失败或定位效果不理想,可以启动手动定位,使用方法是:在图片显示部分鼠标双击左键,即可激活矩形选框,再进行选择就好了。(如果非要选择不是银行卡的部分,那也拿你没办法,你开心就好:) 下面是一些手动定位操作的GIF图:

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2021-07-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 机器学习AI算法工程 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 交互界面
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档