Kera实现AlexNet模型(简易版)

版权声明:转载请注明来源,谢谢。 https://blog.csdn.net/luke_sanjayzzzhong/article/details/89643806

前言

为了掌握更好地掌握深度学习的模型,同时也为了加强一下keras的学习,所以简略复现一下AlexNet。之后还会用tensorflow复现,目前只是想从大体上掌握一下。
论文地址:ImageNet Classification with Deep Convolutional
Neural Networks

模型结构

在这里插入图片描述
感觉这样的图会更容易看清楚一点,因为作者的论文发表的时候GPU的计算力还很低,所以用了两块显卡。我们现在就直接点,直接把模型弄成一张显卡来训练。之前读论文的时候,我一直以为论文里错了,输入明明是224x244x3怎么经过卷积得到的会是55x55x96,今天突然醒悟:原来他做了padding!把输入的224x224的图片padding成了227x227x3。有兴趣的可以到论文中去研究。

猜你喜欢

转载自blog.csdn.net/luke_sanjayzzzhong/article/details/89643806