引言:上篇文章《VGG-net训练过程总结》详细介绍了VGG网络结构和训练过程,主要是使用1.x训练。今天在学习Tensorflow2.0,也从新使用2.0来实现VGG网络。
![tensorflow2.0+数据增强+VGG-Net(私聊领代码)](http://p2.ttnews.xyz/loading.gif)
1、导入相关包,出现红色下波浪线,是pycharm的锅,不影响代码的运行。
![tensorflow2.0+数据增强+VGG-Net(私聊领代码)](http://p2.ttnews.xyz/loading.gif)
导入相关包
2、设置训练参数
设置训练参数
3、定义读取数据集函数
4、定义网络模型
5、定义学习率调整函数,加快网络收敛
6、主函数
不进行数据扩充
进行数据扩充
附代码:有需要代码的朋友可以私聊我领取
tensorflow2.0使用了大量keras的API接口,基本是在keras的基础上套了一个tensorflow的壳子,有一个参考的代码上手非常快。欢迎收藏不看系列同学收藏、转发。
閱讀更多 AI機器學習與數據挖掘 的文章