自学内容网 自学内容网

每天五分钟深度学习框架pytorch:基于vgg块搭建VGG卷积神经网络

本文重点

前面我们使用pytorch搭建了vgg块,本文我们使用vgg块搭建卷积神经网络VGG16,我们先来看一下vgg16的模型结构是什么样的:

搭建vgg16

import torch

from torch import nn

def vgg_block(num_convs,in_channels,out_channels):

net=[nn.Conv2d(in_channels,out_channels,kernel_size=3,padding=1),nn.ReLU(inplace=True)]

for i in range(num_convs-1):

net.append(nn.Conv2d(out_channels,out_channels,kernel_size=3,padding=1)) net.append(nn.ReLU(inplace=


原文地址:https://blog.csdn.net/huanfeng_AI/article/details/145170158

免责声明:本站文章内容转载自网络资源,如侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!