alexnet论文代码复现

一、环境配置
opencv-python 4.1.2.30
Pillow 8.1.2
tensorboard 2.4.1
torch 1.4.0+cuda10.1
torchsummary 1.5.1
torchvision 0.5.0+cuda10.1
numpy 1.20.1
二、项目结构
预训练模型下载链接
https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth
数据集
链接：https://pan.baidu.com/s/1FUiLsw91DW8ssmnbmrxv5Q
提取码：0fgw

alexnet_inference.py：1000类Alexnet测试
alexnet_visualization.py：alexnet卷积核可视化 ，特征图可视化
train_alexnet.py：猫狗数据集上训练Alexnet
三、关键代码

1、torch.topk

torch.topk(input, k, dim