当前位置:   article > 正文

Tensorflow CNN车牌识别_tensorflow crnn车牌

tensorflow crnn车牌

数据集和源代码来自 https://blog.csdn.net/qq_36148847/article/details/79306762

萌新入门,如果过程有误请指正

当时做的时候数据集还只有几百张,今天看发现已经扩充了很多

下面的代码是当时实验的时候的代码,只识别了六个字

原博给的代码经过测试后,发现过拟合问题严重,模型不稳定,因此作以下修改:

1.更改了参数初始化方式:改成xavier初始化方法

参数初始化的目的是为了让神经网络在训练过程中学习到有用的信息,Glorot认为:优秀的初始化应该使得各层的激活值和状态梯度的方差在传播过程中的方差保持一致,Xavier初始化确实保证了Glorot条件。

2.更改了激活函数:改成leaky-relu

ReLU是将所有的负值都设为零,相反,Leaky ReLU是给所有负值赋予一个非零斜率。

3.增加了正则化过程:L2正则化

结果训练集准确率是96.77%左右,当时测试集太小了,准确率算出来没意义。

4.准确率开始下降的时候停止训练,准确率不变的时候停止训练

还有其他修改想起来再继续补……增加了tensorboard监测,不过被我注释掉一些

  1. import sys
  2. import os
  3. import time
  4. import random
  5. import matplotlib.pyplot as plt
  6. import matplotlib.image as ima
  7. import numpy as np
  8. import tensorflow as tf
  9. from PIL import Image
  10. SIZE = 1280
  11. WIDTH = 32
  12. HEIGHT = 40
  13. NUM_CLASSES = 6
  14. iterations = 300
  15. SAVER_DIR = "D:/tf_car_license_dataset/province/"
  16. from tensorflow.contrib.tensorboard.plugins import projector
  17. PROVINCES = ("京","闽","粤","苏","沪","浙")
  18. nProvinceIndex = 0
  19. time_begin = time.time()
  20. # 定义输入节点,对应于图片像素值矩阵集合和图片标签(即所代表的数字)
  21. x = tf.placeholder(tf
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/爱喝兽奶帝天荒/article/detail/851472
推荐阅读
相关标签
  

闽ICP备14008679号