每天五分钟深度学习框架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)!