一、pytorch搭建实战以及sequential的使用
一、pytorch搭建实战以及sequential的使用
- 1.A sequential container
- 2.搭建cifar10 model structure
- 3.创建实例进行测试(可以检查网络是否正确)
- 3.tensorboard图可视化
1.A sequential container
官网说明文档 : https://pytorch.org/docs/stable/generated/torch.nn.Sequential.html#sequential
Example
model = nn.Sequential(nn.Conv2d(1,20,5),nn.ReLU(),nn.Conv2d(20,64,5),nn.ReLU())# Using Sequential with OrderedDict. This is functionally the
# same as the above code
model = nn.Sequential(OrderedDict([('conv1', nn.Conv2d(1,20,5)),('relu1', nn.ReLU()),('conv2', nn.Conv2d(20,64,5)),('relu2', nn.ReLU())]))
2.搭建cifar10 model structure
共有9层网络结构,顺序如下:
(0): Conv2d(3, 32, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2))
(1): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(2): Conv2d(32, 32, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2))
(3): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(4): Conv2d(32, 64, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2))
(5): MaxPool2d(kernel_size=2, stride=2, padding=0, dilation=1, ceil_mode=False)
(6): Flatten(start_dim=1, end_dim=-1)
(7): Linear(in_features=1024, out_features=64, bias=True)
(8): Linear(in_features=64, out_features=10, bias=True)
搭建网络结构:
class Qu(nn.Module):def __init__(self):super(Qu, self).__init__()self.model1 = Sequential(Conv2d(3, 32, 5, padding=2),MaxPool2d(2),Conv2d(32, 32, 5, padding=2),MaxPool2d(2),Conv2d(32, 64, 5, padding=2),MaxPool2d(2),Flatten(),Linear(1024, 64),Linear(64, 10))def forward(self, x):x = self.model1(x)return x
其中
torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode=‘zeros’, device=None, dtype=None)
例如第一个卷积层conv2d的 输入通道=3,输出通道=32,卷积核个数=5x5, padding=2
由于卷积作用后HW不变,仍为32X32,故需计算padding,padding的计算方法如下:
3.创建实例进行测试(可以检查网络是否正确)
qu = Qu()
#print(qu)
input = torch.ones(64, 3, 32, 32)
output = qu(input)
print(output.shape)
输出结果
torch.Size([64, 10])
3.tensorboard图可视化
writer = SummaryWriter("./logs_seq")
writer.add_graph(qu, input)
writer.close()
一、pytorch搭建实战以及sequential的使用相关推荐
- 实战:使用Pytorch搭建分类网络(肺结节假阳性剔除)
实战:使用Pytorch搭建分类网络(肺结节假阳性剔除) 阅前可看: 实战:使用yolov3完成肺结节检测(Luna16数据集)及肺实质分割 其中的脚本资源getMat.py文件是对肺结节进行切割. ...
- Deep Learning:基于pytorch搭建神经网络的花朵种类识别项目(内涵完整文件和代码)—超详细完整实战教程
基于pytorch的深度学习花朵种类识别项目完整教程(内涵完整文件和代码) 相关链接:: 超详细--CNN卷积神经网络教程(零基础到实战) 大白话pytorch基本知识点及语法+项目实战 文章目录 基 ...
- pytorch搭建PyQt5界面实战:ResNet-18实现CLFAR-10图像分类,并进行界面显示
pytorch+PyQt5实战:ResNet-18实现CLFAR-10图像分类,并利用PyQt5进行人机界面显示 实验环境: 1.pytorch-1.6.0 2.python-3.7.9 3.wind ...
- CNN简单实战:PyTorch搭建CNN对猫狗图片进行分类
在上一篇文章:CNN训练前的准备:PyTorch处理自己的图像数据(Dataset和Dataloader),大致介绍了怎么利用pytorch把猫狗图片处理成CNN需要的数据,今天就用该数据对自己定义的 ...
- 深度学习实战(2)用Pytorch搭建双向LSTM
用Pytorch搭建双向LSTM 应最近的课程实验要求,要做LSTM和GRU的实验效果对比.LSTM的使用和GRU十分相似,欢迎参考我的另外一篇介绍搭建双向GRU的Blog:https://blog. ...
- 【深度学习】——利用pytorch搭建一个完整的深度学习项目(构建模型、加载数据集、参数配置、训练、模型保存、预测)
目录 一.深度学习项目的基本构成 二.实战(猫狗分类) 1.数据集下载 2.dataset.py文件 3.model.py 4.config.py 5.predict.py 一.深度学习项目的基本构成 ...
- 【Pytorch神经网络实战案例】21 基于Cora数据集实现Multi_Sample Dropout图卷积网络模型的论文分类
Multi-sample Dropout是Dropout的一个变种方法,该方法比普通Dropout的泛化能力更好,同时又可以缩短模型的训练时间.XMuli-sampleDropout还可以降低训练集和 ...
- Pytorch搭建自己的模型
前言 PyTorch.TensorFlow都是主流的深度学习框架,今天主要讲解一下如何快速使用pytorch搭建自己的模型.至于为什么选择讲解pytorch,这里我就简单说明一下自己的使用感受(相对T ...
- PyTorch搭建预训练AlexNet、DenseNet、ResNet、VGG实现猫狗图片分类
目录 前言 AlexNet DensNet ResNet VGG 前言 在之前的文章中,利用一个简单的三层CNN猫狗图片分类,正确率不高,详见: CNN简单实战:PyTorch搭建CNN对猫狗图片进行 ...
最新文章
- 带你轻而易举的学习python——八皇后问题
- python打开电脑摄像头_Python调用摄像头
- 扬州古城门高挂大红春联迎新春 环卫工受邀揭幕
- c语言冒泡排序(解析)
- Mysql多表查询(案例2)
- Docker 入门(3)Docke的安装和基本配置
- android 6.0 adb,安卓6.0,adb停用系统更新
- 模式识别中的特征向量和矩阵的特征向量有什么关系
- Spring源码之创建AOP代理(补)
- Datatable转实体 实体转换辅助类
- 数据仓库的特点、数据仓库分层、数据仓库模型
- 朋友在B站魔力赏抽到的动漫周边,把我看馋了
- 读书笔记,《刻意练习》,第四章,黄金标准
- Matlab函数参考
- android 模拟器 启动,android开发之启动模拟器并安装游戏apk
- w7系统路由器虚拟服务器怎么设置方法,win7路由器怎么重新设置
- error RC1015: cannot open include file 'sal.h'.
- Vue项目二 登录注册功能的实现
- JavaScript入门 正则表达式/元素运动 Day16
- 无分电器计算机控制的检查方法,无分电器点火系统检修方案.doc
热门文章
- python从零基础到项目实战 当当_Python 3.x网络爬虫从零基础到项目实战
- 99.Shell脚本自启程序并监控(实现全自动监控服务)
- 论文翻译 基于R-FCN的物体检测
- Zabbix4.0 for PostgreSQL (上篇):基础环境的搭建部署
- 详解:Linux Chrony 设置服务器集群同步时间
- Java学习笔记二十六:Java多态中的引用类型转换
- SQL Server误区30日谈-Day27-使用BACKUP ... WITH CHECKSUM可以替代DBCC CheckDB
- [转]论SOA架构的几种主要开发方式
- 软件测试(四)之 PrintPrimes
- 7z(p7zip)压缩软件在Linux下的安装和使用