Layers API介绍
tf.layers包中包含了CNN卷积神经网络的大多数层类型,当前封装支持的层包括:
卷积层
均值池化层
最大池化层
扁平层
密集层
dropout层
BN层
转置卷积层
代码实现
声明输入的占位符
构建网络
基于layers相关层API只需10行代码的卷积网络,包括两个卷积层+两个池化层+两个全链接层+一个输出层。
训练网络
最终在测试集上得到的准确率高达99%以上, 有图为证:
不要因路远而踌躇,
只要去,就必到达!
分享快讯到朋友圈
领取专属 10元无门槛券
私享最新 技术干货