m0_52153205的博客

原创 AlexNet的网络搭建代码与猫狗识别

AlexNet网络是经典神经网络之一，诞生于2012年，我经过网上的资料搜集和研究，整理了利用Alexnet网络模型进行了猫狗分类识别。具体操作如下： AlexNet的网络模型如图所示利用这个模型写出了简单的代码： Net.py import torch from torch import nn import torch.nn.functional as F # 定义一个网络模型 class MyAlexNet(nn.Module): # 初始化网络 def __init__(.