位置: 编程技术 - 正文
使用tensorflow实现AlexNet(tensorflowoom)
编辑：rootadmin推荐整理分享使用tensorflow实现AlexNet(tensorflowoom)，希望有所帮助，仅作参考，欢迎阅读内容。
文章相关热门搜索词:tensorflow怎么调用,tensorflow 例子,怎么用tensorflow,tensorflow 例子,tensorflow用法,怎么用tensorflow,tensorflow tensor操作,tensorflow用法,内容如对您有帮助，希望把文章链接给更多的朋友！
AlexNet是年ImageNet比赛的冠军，虽然过去了很长时间，但是作为深度学习中的经典模型，AlexNet不但有助于我们理解其中所使用的很多技巧，而且非常有助于提升我们使用深度学习工具箱的熟练度。尤其是我刚入门深度学习，迫切需要一个能让自己熟悉tensorflow的小练习，于是就有了这个小玩意儿......
先放上我的代码： map之间都是独立的（除了全连接层），所以这相当于是提升训练速度的一种方法。很多AlexNet的复现都将上下两部分合并了，因为他们都是在单个GPU上运行的。虽然我也是在单个GPU上运行，但是我还是很想将最原始的网络结构还原出来，所以我的代码里也是分开的。
模型定义
定义了卷积、pooling、LRN、dropout、全连接五个模块，其中卷积模块因为将网络的上下两部分分开了，所以比较复杂。接下来定义AlexNet。
buildCNN函数完全按照alexnet的结构搭建网络。loadModel函数从模型文件中读取参数，采用的模型文件见github上的readme说明。至此，我们定义了完整的模型，下面开始测试模型。
模型测试
ImageNet训练的AlexNet有很多类，几乎包含所有常见的物体，因此我们随便从网上找几张图片测试。比如我直接用了之前做项目的渣土车图片：
然后编写测试代码：
如上代码所示，首先需要设置一些参数，然后读取指定路径下的测试图像，再对模型做一个初始化，最后是真正测试代码。测试结果如下：
标签: tensorflowoom
本文链接地址:https://www.jiuchutong.com/biancheng/372303.html 转载请保留说明！
