![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 64
哗啦呼啦嘿
这个作者很懒,什么都没留下…
展开
-
Error-yolov5 pt转onnx报错
针对yolov5 pt转onnx报错进行修改,并根据转换成功的onnx进行预测输出 修改: 去掉Detect层,改为转换后的处理,对三个输出层处理(export grid=True)转换后的onnx的输出少了直接通过detect层输出的[1,6300,9] : 对输出的onnx进行detect层输出并预测(参考链接:基于onnxruntime的YOLOv5单张图片检测实现_wxplol的博客-CSDN博客_onnx yolov5):......原创 2022-07-05 13:22:10 · 1285 阅读 · 0 评论 -
Pytorch-修改模型权重的大小
cfg='models/yolov5s.yaml'device = torch_utils.select_device('0,1')weights = 'weights/qr_model_320.pth'model = Model(cfg).to(device)dict = torch.load(weights)model.load_state_dict(dict,strict=False)print(type(dict)) #<class 'collections.OrderedDi.原创 2020-08-27 17:52:31 · 2653 阅读 · 0 评论 -
pytorch 和 tensorflow-gpu 安装
安装tensorflow-gpu:pip install --user -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.0.0测试(注意:版本不同,有一些细微差别):原创 2020-03-22 22:38:07 · 414 阅读 · 0 评论 -
Pytorch-maskrcnn(seg)前景分割
近来在学习图像分割的相关算法,准备试试看Mask R-CNN的效果。关于Mask R-CNN的详细理论说明,可以参见原作论文https://arxiv.org/abs/1703.06870,网上也有大量解读的文章。本篇博客主要是参考了PyTorch官方给出的训练教程,将如何在自己的数据集上训练Mask R-CNN模型的过程记录下来,希望能为感兴趣的读者提供一些帮助。PyTorch官方教程(...转载 2020-03-20 23:27:17 · 1992 阅读 · 1 评论 -
Pytorch-早停法(early stopping)原理及其代码
作为深度学习训练数据的trick,这个方法必须知道啊,结合交叉验证法,可以防止模型过早拟合。早停法是一种被广泛使用的方法,在很多案例上都比正则化的方法要好。是在训练中计算模型在验证集上的表现,当模型在验证集上的表现开始下降的时候,停止训练,这样就能避免继续训练导致过拟合的问题。其主要步骤如下:1. 将原始的训练数据集划分成训练集和验证集2. 只在训练集上进行训练,并每隔一个周期计算模型在验...原创 2019-12-04 16:02:59 · 51190 阅读 · 13 评论 -
pytorch-将网络模型预测结果的概率值存放到csv文件中
import csvimport cv2#stag_01_submitcsvFile = open("test_submit.csv", "w") #创建csv文件writer = csv.writer(csvFile) #创建写的对象#先写入columns_name writer.writerow(["id","lab...原创 2019-11-29 16:41:29 · 8234 阅读 · 5 评论 -
pytorch-模型训练-加载图像分类模型训练参数
将训练好的模型参数进行测试,检测其正确率,并且写入csv文件中,其每个类的概率值,本模型为图像分类模型:import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 按照PCI_BUS_ID顺序从0开始排列GPU设备 os.environ["CUDA_VISIBLE_DEVICES"] = "1" #设置当前使用的GPU设备仅为0号...原创 2019-11-15 10:07:15 · 877 阅读 · 1 评论 -
pytorch-将保存的模型进行新的图片分类
from PIL import Imageimport torchvision.transforms as Tfrom torch.autograd import Variable as Vimport torch as t model = model.cuda()#导入网络模型model.eval()model.load_state_dict(t.load('./models/n...原创 2019-11-29 16:28:17 · 550 阅读 · 0 评论 -
resnet系列在imagenet上预训练的pytorch模型下载地址
model_urls = { 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', 'resnet50': 'https://downl...转载 2019-07-18 10:24:09 · 9454 阅读 · 9 评论 -
Pytorch-error
from ignite.engine import Engine, EventsImportError: No module named 'ignite.engine'解决如下:pip install pytorch-igniteconda install ignite -c pytorch如果还不行,下载GitHub下的ignite:https://github.c...原创 2019-05-23 10:14:23 · 3469 阅读 · 2 评论 -
Pytorch-Loss Function
本文总结Pytorch中的Loss Function Loss Function是深度学习模型训练中非常重要的一个模块,它评估网络输出与真实目标之间误差,训练中会根据这个误差来更新网络参数,使得误差越来越小;所以好的,与任务匹配的Loss Function会得到更好的模型。 但本文不会讨论什么任务用什么损失函数,只是总结下Pytorch中的Loss Function文章目录Lo...转载 2019-04-11 15:05:57 · 895 阅读 · 0 评论 -
Pytorch-数据类型转换
1.Pytorch上的数据类型Pytorch的类型可以分为CPU和GPU上的Tensor, 它们拥有的数据类型是基本上是一样的:tensor.FloatTensortensor.LongTensortensor.ByteTensortensor.CharTensortensor.ShortTensortensor.IntTensortorch.LongTensor其中torch....转载 2019-03-06 10:37:09 · 23071 阅读 · 1 评论 -
pytorch-detach与detach_
detach官方文档中,对这个方法是这么介绍的。返回一个新的 从当前图中分离的 Variable。 返回的 Variable 永远不会需要梯度 如果 被 detach 的Variable volatile=True, 那么 detach 出来的 volatile 也为 True 还有一个注意事项,即:返回的 Variable 和 被 detach 的Variable 指向同一个 t...转载 2019-03-08 14:55:20 · 1902 阅读 · 0 评论 -
Tensorflow Pytorch GPU安装(Ubuntu 16.04 anaconda cuda8.0 cuDNN6.0)
1. 安装python(Anaconda)python环境使用anaconda从官方网站下载操作系统对应的版本chmod +x Anaconda3-5.0.1-Linux-x86_64.sh./Anaconda3-5.0.1-Linux-x86_64.sh 2. Nvidia显卡驱动(使用NVIDA驱动)重启计算机 3. Nvidia C...转载 2018-11-30 14:41:05 · 773 阅读 · 0 评论 -
各类损失函数-pytorch
转载地址:https://www.jishux.com/p/2a9ae1556486f8be基类定义pytorch损失类也是模块的派生,损失类的基类是_Loss,定义如下class _Loss(Module): def __init__(self, size_average=None, reduce=None, reduction='elementwise_mean'): ...转载 2019-01-02 12:28:57 · 1111 阅读 · 0 评论 -
pytorch-error:CondaHTTPError
pytorch-error:CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/linux-64/pytorch-1.0.0-py3.6_cuda8.0.61_cudnn7.1.2_1.tar.bz2>Elapsed: -An HTTP error occ...原创 2019-01-07 18:05:54 · 2225 阅读 · 2 评论 -
Pytorch_tutorial
视频课程code:https://github.com/dragen1860/DeepLearningTutorialspytorch论文中代码学习:https://blog.csdn.net/Chen_yuazzy/article/details/81002507pytorch升级 conda不行就改成pip:pip3 install https://download.pytorc...原创 2019-03-06 10:46:59 · 367 阅读 · 1 评论 -
Pytorch-模型的保存和导入
在PyTorch中如何保存和恢复模型并查看参数在模型完成训练后,我们需要将训练好的模型保存为一个文件供测试使用,或者因为一些原因我们需要继续之前的状态训练之前保存的模型,那么如何在PyTorch中保存和恢复模型呢?参考PyTorch官方的这份repo,我们知道有两种方法可以实现我们想要的效果。方法一(推荐):第一种方法也是官方推荐的方法,只保存和恢复模型中的参数。保存...转载 2019-02-23 16:31:48 · 4972 阅读 · 2 评论 -
CUDA-Pinned Host Memory
翻译原文:https://devblogs.nvidia.com/how-optimize-data-transfers-cuda-cc/Pinned Host MemoryHost (CPU) data allocations are pageable by default. The GPU cannot access data directly from pageable host m...翻译 2019-02-23 19:19:01 · 2427 阅读 · 0 评论 -
Pytorch-Compose(transforms)
pytorch torchvision transform对PIL.Image进行变换class torchvision.transforms.Compose(transforms)将多个transform组合起来使用。transforms: 由transform构成的列表. 例子:transforms.Compose([ transforms.CenterCrop...转载 2019-02-25 16:52:43 · 21293 阅读 · 3 评论 -
Pytorch-view函数
相当于numpy中resize()的功能,但是用法可能不太一样。把原先tensor中的数据按照行优先的顺序排成一个一维的数据(这里应该是因为要求地址是连续存储的),然后按照参数组合成其他维度的tensor。比如说是不管你原先的数据是[[[1,2,3],[4,5,6]]]还是[1,2,3,4,5,6],因为它们排成一维向量都是6个元素,所以只要view后面的参数一致,得到的结果都是一样的。比如,...转载 2019-02-25 18:52:03 · 5655 阅读 · 0 评论 -
pytorch入门教程:构造一个小型CNN
pytorch入门教程:构造一个小型CNN 学过深度卷积网络的应该都非常熟悉这张demo图(LeNet): 此图是LeNet的结构图,把32*32的手写英文字符图片作为输入,训练出一个对于手写字符的分类器我们训练这个网络必须经过4步:第一步:将输入input向前传播,进行运算后得到输出output第二步:将output再输入loss函数,计算loss值(是个标量)第...转载 2018-11-09 19:40:29 · 3685 阅读 · 0 评论