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

本文重点

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

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

搭建vgg16

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

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=

你可能感兴趣的:(深度学习框架pytorch,深度学习,pytorch,cnn,VGG,卷积神经网络)