Mitchy
今日もキラキラしていこう♪

实战神经网络项目,识别图像中有无信号灯以及什么颜色的信号灯

扩展阅读:http://yann.lecun.com/exdb/publis/pdf/sermanet-ijcnn-11.pdf

利用了LeNet的结构来对交通信号灯进行分类

步骤:

导入数据

初始化,打印看看

预处理(正则化,打乱顺序,分出验证集)

设置5层的LeNet网络

  CNN + ReLU

  CNN + ReLU

  Max pooling + flattern

  Fully connected + ReLU

  Fully connected + ReLU

  Fully connected

初始化训练所需函数,定义训练过程

跑!

 

查看自己模型训练的准确率和测试的准确率,来判断训练得怎么样

用tf.nn.top_k来查看自己模型前k个(k应该小于5)分类判断的概率

  

posted on 2020-09-13 21:56  Mitchy  阅读(152)  评论(0编辑  收藏  举报