TensorFlow搭建卷积网络识别交通标志

目标是修改LeNet用于识别德国交通标志,共43个类别。LeNet有两个卷积层、三个全链接层,我增长了一层卷积,修改了kernel大小。另外,在全链接层中加入了dropou用于防止过拟合。这里记录下搭建网络的过程,总结使用TensorFlow搭建并训练卷积网络的方法。python 1. 数据集的载入、augment与normalization 载入数据集 """ 载入数据集 """ # 下载好的数
相关文章
相关标签/搜索