alexnet和vgg网络的一些个人看法

从2012年之后,alexnet网络的横空出世,带来了深度学习的春天,深度学习在图像,语音等领域大放异彩,几乎碾压其它传统机器学习算法,之后的vgg网络更是被用于计算机视觉的各个任务中,比如本人研究的reid方向,很多方法都是基于vgg网络来设计算法,下面就简单介绍一下这两个网络的优秀之处。

1,alexnet

lenet是最早用于数字识别的cnn网络,alexnet也主要是对于此网络的改进,下面直接看pytorch对于这2个网络的代码实现:

1.1 lenet

这里写图片描述


class LeNet5(nn.Module):
    def __init__(self):
        super().__init__()
        #定义卷积层,1个输入通道,6个输出通道,5*5的卷积filter,外层补上了两圈0,因为输入的是32*32
        self.conv1 = nn.Conv2d(1, 6, 5, padding=2)
        #第二个卷积层,6个输入,16个输出,5*5的卷积filter 
        self.conv2 = nn.Conv2d(6, 16, 5)

        #最后是三个全连接层
        self.fc1 = nn.Linear(16*5*5, 120)
        self.fc2 = nn.Linear(120, 84)
   

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值