PyTorch搭建VGG网络

最新推荐文章于 2024-06-25 09:30:14 发布

Tra_cy

最新推荐文章于 2024-06-25 09:30:14 发布

阅读量2.4k

点赞数 4

分类专栏： pytorch学习

本文链接：https://blog.csdn.net/Tra_cy/article/details/103269899

版权

本文介绍了如何在PyTorch中搭建VGG19网络，包括网络结构、构建模型的步骤以及模型检验的方法，涉及torch.nn.Module、Sequential、MaxPool2d、Conv2d等组件的使用。

摘要由CSDN通过智能技术生成

1. VGG Net 网络结构

在这里插入图片描述

2. 搭建过程（以VGG19为例）

（1）加载必要及准备工作

import torch.nn as nn
import torch.utils.model_zoo as model_zoo
import torch
from torch.nn import functional as F

#对外暴露接口，当导入该模块时，只导入枚举列表里面的
__all__ = ['vgg19']
#预训练模型下载地址
model_urls = {
   
    'vgg19': 'https://download.pytorch.org/models/vgg19-dcbb9e9d.pth',
}

# 可以定义不同的vgg结构，这样写可以有效节约代码空间，下面以vgg19为例
cfg = {
   
    'E': [64, 64, 'M', 128, 128, 'M', 256, 256, 256, 256, 'M', 512, 512, 512, 512, 'M'