由于基础不是很好,最近花了大概五天的时间。终于把Alexnet网络实现了。首先看一下网络结构。

test,test2分别是训练网络时生成的文件。input_data是处理数据的文件,3个model文件,由上至下分别对应了最原始,最简单的Alexnet网络结构,其次是模仿2块GPU进行训练的网络结构,最后是去掉lrn层的网络结构。test.py文件是取一张没有训练过的图片进行测试,两个training文件对应两个model文件的网络进行训练。我并没有训练去掉lrn层的网络,感兴趣的朋友也可以将training.py改动一下进行训练。tes