实战神经网络项目,识别图像中有无信号灯以及什么颜色的信号灯
扩展阅读: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)分类判断的概率