pytorch
文章平均质量分 83
介绍pytorch的使用
weixin_44457930
这个作者很懒,什么都没留下…
展开
-
OpenVINO2023使用简介
OpenVINO使用简介原创 2023-08-29 22:56:49 · 1785 阅读 · 0 评论 -
onnx模型的保存及其在OpenCV中使用
pytorch模型部署到OpenCV环境中原创 2023-08-01 22:04:39 · 1972 阅读 · 0 评论 -
pytorch搭建深度学习模型,哪些步骤占用显存
pytorch的显存机制,一直不明白,今天看了两篇文章,终于有些明白了,这两篇文章如下:https://zhuanlan.zhihu.com/p/424512257https://blog.csdn.net/qq_43827595/article/details/115722953根据第一篇知乎文章及其评论,我自己写了一套代码来记录各个步骤中显存的占用情况,以下是通过jupyter notebook中转化而来得到的markdown文件内容:import torchimport torch.nn a原创 2022-03-02 21:00:18 · 1179 阅读 · 1 评论 -
关于pytorch多线程导入数据时的随机问题,pytorch遇到问题时的网上求助流程
在“从零开始实现yolox一:数据集类”的测试代码中,使用了多线程导入数据当使用多线程之后,即便在程序开头使用了np.random.seed(0)来设置种子,但每次得到的结果都不一样。程序如下:import numpy as npimport matplotlib.pyplot as pltimport matplotlib.patches as patchesimport torchfrom torch.utils.data import DataLoaderfrom utils.da原创 2022-02-19 11:55:27 · 1102 阅读 · 0 评论 -
RTX3060安装pytorch
RTX3060安装pytorch1 安装anaconda2 查看本机显卡支持的cuda最高版本(2)根据pytorch版本选择要安装的CUDA(3)下载安装CUDA(4)下载cudNN(5)下载安装刚刚选择的pytorch版本前不久刚刚入手了一台新电脑,显卡为RTX3060,在安装环境的时候,踩了不少坑,现在将经验总结如下:1 安装anaconda这个可以看这个教程:https://blog.csdn.net/in546/article/details/117400839需要注意的是,要记得添加到环原创 2022-01-26 19:09:46 · 7205 阅读 · 5 评论 -
pytorch 建立LSTM模型实现股价预测
时间序列模型LSTM的建模与数据处理过程转载 2022-01-22 16:35:31 · 11200 阅读 · 31 评论 -
从头开始复现YOLOv3(三)训练模型
YOLOv3模型训练1 迁移学习(1)两种权重文件(2)导入权重方法(3)保存模型的方法2 标签转化函数3 模型训练4.模型评价(1)mAP的计算原理(2)mAP的计算程序(3)在训练中添加模型的评价前面我们已经讲解了模型的搭建和数据类,现在可以来正式训练模型了。1 迁移学习(1)两种权重文件我们基本不会去从0开始训练模型,而是从官网把已经训练好的模型导入,即迁移学习。在yolo3_from_scratch中新建一个名为weights的文件夹,将权重文件放入,权重文件有两种,一种是整个模型的参数文原创 2021-12-13 12:00:49 · 2648 阅读 · 0 评论 -
pytorch第07天 迁移学习实验
迁移学习1 pytorch中迁移学习概述2 建立一个预训练的模型文件与参数文件2 实验实验1:不写类,能否直接导入模型文件和参数文件实验2:类名不一致,但组件名称一致,能否导入参数文件实验3:类名不一致,不带参数的组件名称不一致,但带参数的组件一致,能否导入参数文件实验4:带参数的组件名称不一致,能否导入参数文件实验5:定义类之后,类名不一致,但各组件名称一致,能否导入模型文件实验6:定义类之后,类名一致,但组件名称不一致,能否导入模型文件总结1 pytorch中迁移学习概述所谓迁移学习,就是将已经预训原创 2021-08-27 12:37:36 · 588 阅读 · 0 评论 -
pytorch第06天 图片分类器
图片分类器1 数据(1)数据集介绍与导入(2)transforms.Compose与transforms.Normalize(3)图片预览1 数据(1)数据集介绍与导入对于视觉任务,可以通过torchvision模块导入公共数据集和其上的操作,这里我们以torchvision.datasets.CIFAR10作为数据集,训练图片分类器。CIFAR10中的图像大小为3x32x32,即32x32像素的3通道彩色图像,数据集有10个类别,可以看下面的示意图我们现在把数据集导入进来import tor原创 2021-08-19 16:35:49 · 555 阅读 · 0 评论 -
pytorch第05天 搭建神经网路
搭建第一个神经网络1 数据(1)导入数据(2)数据集可视化1 数据(1)导入数据我们以Fashion-MNIST数据集为例,介绍一下关于pytorch的数据集导入。PyTorch域库提供许多预加载的数据集(如FashionList),这些数据集为torch.utils.data.Dataset的子类并实现特定于特定数据的功能。Fashion-MNIST是Zalando文章中的图像数据集,包含60,000个训练示例和10,000个测试示例。每个示例包括28×28灰度图像和来自10个类中的一个的关联标原创 2021-08-17 22:00:42 · 1867 阅读 · 3 评论 -
pytorch第04天 梯度下降
梯度下降1 激活函数2 损失函数3 梯度4 自动求导(1)通过torch.autograd.grad获得梯度(2)通过反向传播,将梯度复制到需要求导的变量上5 保留计算图6 softmax(1)softmax过程(2)softmax函数求导(3)softmax的求导程序1 激活函数神经网络常用的激活函数有三种:sigmoid、tanh、reluimport torchfrom torch.nn import functional as Fz = torch.linspace(-1, 1, 6)原创 2021-08-15 11:04:09 · 258 阅读 · 0 评论 -
pytorch第03天 索引与切片
1 索引看下面的程序,重点看注释,就能明白索引什么意思import torcha = torch.rand(2, 3, 4)print('---------a------------')print(a)print(a.size())print('--------a[0]----------')print(a[0]) # 表示第一个维度取0,其他维度没有标明,表示全选print(a[0].size())print('--------a[0, 1]---------')print原创 2021-06-17 21:08:38 · 176 阅读 · 1 评论 -
pytorch第02天 创建张量
创建张量1 rand,randn,randint,rand_like,rand_perm的使用2 张量类型与元素类型的关系3 填充函数4 arange与range函数5 randn(shape) 生成标准正态分布6 linspace7 单位矩阵8 今日重点除了可以像上节课一样,使用.tensor(),.FloatTensor(),.Tensor()等函数来初始化一个张量,还可以用.from_numpy()函数把numpy的数据类型转化为tensor(详见第一天笔记),还可以用很多方法生成。1 rand原创 2021-06-17 20:54:02 · 525 阅读 · 1 评论 -
pytorch第01天 数据类型
数据类型1 pytorch中的常用数据类型2 数据的尺度与维度(1)0维数据(2)1维数据(3)2维数据3 获取数据的维度数、形状、元素个数等信息4 用.type()返回张量的数据类型5 初始化张量.tensor(),.FloatTensor(),.Tensor()的区别6 张量的长度7 Python知识补充(1)is instance()(2)iter()函数和next()函数1 pytorch中的常用数据类型32位浮点型:torch.FloatTensor。64位整型:torch.LongTens原创 2021-06-17 11:42:46 · 1095 阅读 · 2 评论