tensorflow2.0+数据增强+VGG-Net(私聊领代码)

引言:上篇文章《VGG-net训练过程总结》详细介绍了VGG网络结构和训练过程,主要是使用1.x训练。今天在学习Tensorflow2.0,也从新使用2.0来实现VGG网络。

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

1、导入相关包,出现红色下波浪线,是pycharm的锅,不影响代码的运行。


tensorflow2.0+数据增强+VGG-Net(私聊领代码)

导入相关包

2、设置训练参数

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

设置训练参数

3、定义读取数据集函数

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

4、定义网络模型

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

5、定义学习率调整函数,加快网络收敛

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

6、主函数

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

不进行数据扩充

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

进行数据扩充

tensorflow2.0+数据增强+VGG-Net(私聊领代码)

附代码:有需要代码的朋友可以私聊我领取

tensorflow2.0使用了大量keras的API接口,基本是在keras的基础上套了一个tensorflow的壳子,有一个参考的代码上手非常快。欢迎收藏不看系列同学收藏、转发。


分享到:


相關文章: