一、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的使用相关推荐

  1. 实战:使用Pytorch搭建分类网络(肺结节假阳性剔除)

    实战:使用Pytorch搭建分类网络(肺结节假阳性剔除) 阅前可看: 实战:使用yolov3完成肺结节检测(Luna16数据集)及肺实质分割 其中的脚本资源getMat.py文件是对肺结节进行切割. ...

  2. Deep Learning:基于pytorch搭建神经网络的花朵种类识别项目(内涵完整文件和代码)—超详细完整实战教程

    基于pytorch的深度学习花朵种类识别项目完整教程(内涵完整文件和代码) 相关链接:: 超详细--CNN卷积神经网络教程(零基础到实战) 大白话pytorch基本知识点及语法+项目实战 文章目录 基 ...

  3. 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 ...

  4. CNN简单实战:PyTorch搭建CNN对猫狗图片进行分类

    在上一篇文章:CNN训练前的准备:PyTorch处理自己的图像数据(Dataset和Dataloader),大致介绍了怎么利用pytorch把猫狗图片处理成CNN需要的数据,今天就用该数据对自己定义的 ...

  5. 深度学习实战(2)用Pytorch搭建双向LSTM

    用Pytorch搭建双向LSTM 应最近的课程实验要求,要做LSTM和GRU的实验效果对比.LSTM的使用和GRU十分相似,欢迎参考我的另外一篇介绍搭建双向GRU的Blog:https://blog. ...

  6. 【深度学习】——利用pytorch搭建一个完整的深度学习项目(构建模型、加载数据集、参数配置、训练、模型保存、预测)

    目录 一.深度学习项目的基本构成 二.实战(猫狗分类) 1.数据集下载 2.dataset.py文件 3.model.py 4.config.py 5.predict.py 一.深度学习项目的基本构成 ...

  7. 【Pytorch神经网络实战案例】21 基于Cora数据集实现Multi_Sample Dropout图卷积网络模型的论文分类

    Multi-sample Dropout是Dropout的一个变种方法,该方法比普通Dropout的泛化能力更好,同时又可以缩短模型的训练时间.XMuli-sampleDropout还可以降低训练集和 ...

  8. Pytorch搭建自己的模型

    前言 PyTorch.TensorFlow都是主流的深度学习框架,今天主要讲解一下如何快速使用pytorch搭建自己的模型.至于为什么选择讲解pytorch,这里我就简单说明一下自己的使用感受(相对T ...

  9. PyTorch搭建预训练AlexNet、DenseNet、ResNet、VGG实现猫狗图片分类

    目录 前言 AlexNet DensNet ResNet VGG 前言 在之前的文章中,利用一个简单的三层CNN猫狗图片分类,正确率不高,详见: CNN简单实战:PyTorch搭建CNN对猫狗图片进行 ...

最新文章

  1. 带你轻而易举的学习python——八皇后问题
  2. python打开电脑摄像头_Python调用摄像头
  3. 扬州古城门高挂大红春联迎新春 环卫工受邀揭幕
  4. c语言冒泡排序(解析)
  5. Mysql多表查询(案例2)
  6. Docker 入门(3)Docke的安装和基本配置
  7. android 6.0 adb,安卓6.0,adb停用系统更新
  8. 模式识别中的特征向量和矩阵的特征向量有什么关系
  9. Spring源码之创建AOP代理(补)
  10. Datatable转实体 实体转换辅助类
  11. 数据仓库的特点、数据仓库分层、数据仓库模型
  12. 朋友在B站魔力赏抽到的动漫周边,把我看馋了
  13. 读书笔记,《刻意练习》,第四章,黄金标准
  14. Matlab函数参考
  15. android 模拟器 启动,android开发之启动模拟器并安装游戏apk
  16. w7系统路由器虚拟服务器怎么设置方法,win7路由器怎么重新设置
  17. error RC1015: cannot open include file 'sal.h'.
  18. Vue项目二 登录注册功能的实现
  19. JavaScript入门 正则表达式/元素运动 Day16
  20. 无分电器计算机控制的检查方法,无分电器点火系统检修方案.doc

热门文章

  1. python从零基础到项目实战 当当_Python 3.x网络爬虫从零基础到项目实战
  2. 99.Shell脚本自启程序并监控(实现全自动监控服务)
  3. 论文翻译 基于R-FCN的物体检测
  4. Zabbix4.0 for PostgreSQL (上篇):基础环境的搭建部署
  5. 详解:Linux Chrony 设置服务器集群同步时间
  6. Java学习笔记二十六:Java多态中的引用类型转换
  7. SQL Server误区30日谈-Day27-使用BACKUP ... WITH CHECKSUM可以替代DBCC CheckDB
  8. [转]论SOA架构的几种主要开发方式
  9. 软件测试(四)之 PrintPrimes
  10. 7z(p7zip)压缩软件在Linux下的安装和使用