我们已经为您准备好复现该模型所需的环境,您可以创建实例、一键复用哦!

模型的使用文档涵盖模型所需环境、模型简介以及复现模型的具体步骤。模型效果和模型详细来源可以点击论文地址和模型来源链接查看哦~

AlexNet

2022-01-25 model:alexnet2.0-ubuntu18.04-cuda10.1-python3.7-tensorflow2.3.0-new

模型标签
图像分类

模型简介

模型介绍:AlexNet模型是Alex和Hinton参加ILSVRC2012比赛的卷积网络论文,本网络结构也是开启ImageNet数据集更大,更深CNN的开山之作,本文对CNN的一些改进成为以后CNN网络通用的结构;在一些报告中被称为Alex-net,之后在Imagenet上取得更好结果的ZF-net,SPP-net,VGG等网络,都是在其基础上修改得到。 论文地址https://proceedings.neurips.cc/paper/2012/file/c399862d3b9d6b76c8436e924a68c45b-Paper.pdf



1. 使用方法

1.1. 模型训练

进入终端,在root/AlexNet(默认)路径下,运行以下命令:

python train.py

1.2. 模型测试

进入终端,在root/AlexNet(默认)路径下,运行以下命令:

python test.py

2. 模型来源

https://github.com/dansuh17/alexnet-pytorch.git