AlexNet+VGGNet模型框架（PyTorch）

最新推荐文章于 2024-01-22 12:19:38 发布

Leung WaiHo

最新推荐文章于 2024-01-22 12:19:38 发布

阅读量189

点赞数

分类专栏：常见CNN模型框架文章标签：深度学习卷积神经网络 cv

本文链接：https://blog.csdn.net/weixin_41978493/article/details/107676971

版权

常见CNN模型框架专栏收录该内容

4 篇文章 1 订阅

订阅专栏

I. 前言

AlexNet，论文地址：ImageNet Classification with Deep Convolutional Neural Networks.
VGGNet，论文地址：Very Deep Convolutional Networks for Large-Scale Image Recognition. 论文中提出了六种模型，其中较为有名的即是VGG-16及VGG-19.

II. 模型构架图

1. AlexNet

在这里插入图片描述

2. VGGNet

在这里插入图片描述
【注】最后两列分别是VGG-16及VGG-19，标黄的部分为与前者在构架上的不同之处.

III. 代码复现

1. Alexnet

import torch.nn as nn
class Alexnet(nn.Module):
    def __init__(self, num_classes):
        super(Alexnet, self).__init__()
        #特征提取
        self.features = nn.Sequential(
            nn.Conv2d(3, 96, kernel_size=11, stride=4, padding=2),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
        
            nn.Conv2d(96, 256, kernel_size=5, stride=1, padding=2),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
        
            nn.Conv2d(256, 384, kernel_size=3, stride=1, padding=1),
            nn.ReLU(inplace=True),
            
            nn.Conv2d(384, 384, kernel_size=3, stride=1, padding=1),
            nn.ReLU(inplace=True),
            
            nn.Conv2d(384, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
        )
        
        self.classifier = nn.Sequential(
            nn.Dropout(),
            nn.Linear(256, 6, 6, 4096),
            nn.ReLU(inplace=True),
            
            nn.Dropout(),
            nn.Linear(4096, 4096),
            nn.ReLU(inplace=True),
            
            nn.Linear(4096, num_classes))
        
    def forward(self, x):
        x = self.features(x)
        x = view(x.size(0), 256*6*6)
        x = self.classifier(x)
        
        return x

2. VGG-16

class VGG16net(nn.Module):
    def __init__(self):
        super(VGGnet, self).__init__()
        self.features = nn.Sequential(
            nn.Conv2d(3, 64, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(64, 64, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(64, 128, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(128, 128, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(128, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(256, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
        )
        
        self.classifier = nn.Sequential(
            nn.Dropout(),
            nn.Linear(7*7*512, 4096),
            nn.ReLU(True),
            
            nn.Dropout(),
            nn.Linear(4096, 4096),
            nn.ReLU(True),
            
            nn.Linear(4096, 1000),
        )
        
    def forward(self, x):
        x = self.features(x)
        x = view(x.size(0), 7*7*512)
        x = self.classifier(x)
        return x

3. VGG-19

class VGG19net(nn.Module):
    def __init__(self):
        super(VGGnet, self).__init__()
        self.features = nn.Sequential(
            nn.Conv2d(3, 64, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(64, 64, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(64, 128, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(128, 128, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(128, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(256, 256, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(256, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
            
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.Conv2d(512, 512, kernel_size=3, stride=1, padding=1),
            nn.ReLU(True),
            nn.MaxPool2d(kernel_size=2, stride=2),
        )
        
        self.classifier = nn.Sequential(
            nn.Dropout(),
            nn.Linear(7*7*512, 4096),
            nn.ReLU(True),
            
            nn.Dropout(),
            nn.Linear(4096, 4096),
            nn.ReLU(True),
            
            nn.Linear(4096, 1000),
        )
        
    def forward(self, x):
        x = self.features(x)
        x = view(x.size(0), 7*7*512)
        x = self.classifier(x)
        return x

Leung WaiHo

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
AlexNet+VGGNet模型框架（PyTorch）

I. 前言AlexNet，论文地址：ImageNet Classification with Deep Convolutional Neural Networks.VGGNet，论文地址：Very Deep Convolutional Networks for Large-Scale Image Recognition. 论文中提出了六种模型，其中较为有名的即是VGG-16及VGG-19.II. 模型构架图AlexNet-VGGNet【注】最后两列分别是VGG-16及VGG-19，标黄
复制链接

扫一扫

专栏目录