草庐IT

pytorch-lightning

全部标签

Pytorch使用Grad-CAM绘制热力图

原理与代码学习自B站霹雳吧啦Wz老师使用grad_cam对不同预测目标的图像做activate图。效果见下图。使用的是自己训练的MobileNetV2需要模型feature的最后一层,模型训练权重。代码如下:importjsonimportosimportnumpyasnpimporttorchfromPILimportImageimportmatplotlib.pyplotaspltfromtorchvisionimportmodelsfromtorchvisionimporttransformsfromutilsimportGradCAM,show_cam_on_image,center_

如何在PyCharm中搭建PyTorch环境并训练模型?

一、前言PyTorch是由Facebook开源的深度学习框架,它具有易用、高效、灵活等特点,被广泛应用于学术界和工业界中。而PyCharm是JetBrains公司开发的一款强大的PythonIDE,它可以帮助开发者提高开发效率,提供了丰富的代码编辑、调试、测试以及版本控制等功能。本文将介绍如何在PyCharm中搭建PyTorch开发环境。二、安装PyCharm首先需要下载并安装PyCharm,可以在JetBrains官网中下载PyCharm的社区版或专业版,本文以社区版为例。下载地址:https://www.jetbrains.com/pycharm/download/安装过程中需要选择安装路

Pytorch报错解决——(亲测有效)RuntimeError: Distributed package doesn‘t have NCCL built in

在我跑SLADD模型的时候,出现了如下报错:上网搜寻一番后,发现了解决方法第一步:在本文件的开头机上这样两行代码:importosos.environ["PL_TORCH_DISTRIBUTED_BACKEND"]="gloo"第二步:在本文件代码中找到这样一部分代码:把dist.init_process_group(backend='nccl',init_method='env://',world_size=args.world_size,rank=rank)这一行中的nccl换成gloo若要解释原因就是windows系统不支持NCCL,只能换用GLOO亲测这个报错就没了,然而接下来还有好多

再探pytorch的Dataset和DataLoader

本文参加新星计划人工智能(Pytorch)赛道:https://bbs.csdn.net/topics/613989052本文从分类、检测、分割三大任务的角度来剖析pytorch得dataset和dataloader源码,可以让初学者深刻理解每个参数的由来和使用,并轻松自定义dataset。思考:在探究Dataset和DataLoader之前,需要明白一个事情,就是当我们不管做是分类、检测还是分割任务时,我们的数据集一定由很多张图片组成的,形状大小各异;那麽我们在使用pytorch时,图片是怎么以一个batch的形式进行打包的呢,形状不同怎么处理,数据格式有什么要求,Dataset类中的初始化

再探pytorch的Dataset和DataLoader

本文参加新星计划人工智能(Pytorch)赛道:https://bbs.csdn.net/topics/613989052本文从分类、检测、分割三大任务的角度来剖析pytorch得dataset和dataloader源码,可以让初学者深刻理解每个参数的由来和使用,并轻松自定义dataset。思考:在探究Dataset和DataLoader之前,需要明白一个事情,就是当我们不管做是分类、检测还是分割任务时,我们的数据集一定由很多张图片组成的,形状大小各异;那麽我们在使用pytorch时,图片是怎么以一个batch的形式进行打包的呢,形状不同怎么处理,数据格式有什么要求,Dataset类中的初始化

Pytorch实现GCN(基于PyTorch实现)

文章目录前言一、导入相关库二、加载Cora数据集三、定义GCN网络3.1定义GCN层3.1.1邻接矩阵A3.1.2度矩阵D3.1.3获取矩阵D−1/2D^{-1/2}D−1/23.1.4GCNConv层3.2定义GCN网络四、定义模型五、模型训练六、模型验证七、结果完整代码前言大家好,我是阿光。本专栏整理了《图神经网络代码实战》,内包含了不同图神经网络的相关代码实现(PyG以及自实现),理论与实践相结合,如GCN、GAT、GraphSAGE等经典图网络,每一个代码实例都附带有完整的代码。正在更新中~✨🚨我的项目环境:平台:Windows10语言环境:python3.7编译器:PyCharmPy

Pytorch实现GCN(基于PyTorch实现)

文章目录前言一、导入相关库二、加载Cora数据集三、定义GCN网络3.1定义GCN层3.1.1邻接矩阵A3.1.2度矩阵D3.1.3获取矩阵D−1/2D^{-1/2}D−1/23.1.4GCNConv层3.2定义GCN网络四、定义模型五、模型训练六、模型验证七、结果完整代码前言大家好,我是阿光。本专栏整理了《图神经网络代码实战》,内包含了不同图神经网络的相关代码实现(PyG以及自实现),理论与实践相结合,如GCN、GAT、GraphSAGE等经典图网络,每一个代码实例都附带有完整的代码。正在更新中~✨🚨我的项目环境:平台:Windows10语言环境:python3.7编译器:PyCharmPy

UserWarning: __floordiv__is deprecated, and its behavior will change in a future version of pytorch.

报错提示:UserWarning:__floordiv__isdeprecated,anditsbehaviorwillchangeinafutureversionofpytorch.Itcurrentlyroundstoward0(likethe‘trunc’functionNOT‘floor’).Thisresultsinincorrectroundingfornegativevalues.Tokeepthecurrentbehavior,usetorch.div(a,b,rounding_mode=‘trunc’),orforactualfloordivision,usetorch.di

Pytorch机器学习(八)—— YOLOV5中NMS非极大值抑制与DIOU-NMS等改进

Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进目录Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进前言一、NMS非极大值抑制算法二、Hard-NMS非极大值代码三、DIOU-NMS 四、soft-NMS前言在目标检测的预测阶段时,会输出许多候选的anchorbox,其中有很多是明显重叠的预测边界框都围绕着同一个目标,这时候我就可以使用NMS来合并同一目标的类似边界框,或者说是保留这些边界框中最好的一个。如果对IOU等知识不了解的可以看我上篇博客Pytorch机器学习(五)——目标检测中的损失函数(l2,IOU,G

Pytorch机器学习(八)—— YOLOV5中NMS非极大值抑制与DIOU-NMS等改进

Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进目录Pytorch机器学习(八)——YOLOV5中NMS非极大值抑制与DIOU-NMS等改进前言一、NMS非极大值抑制算法二、Hard-NMS非极大值代码三、DIOU-NMS 四、soft-NMS前言在目标检测的预测阶段时,会输出许多候选的anchorbox,其中有很多是明显重叠的预测边界框都围绕着同一个目标,这时候我就可以使用NMS来合并同一目标的类似边界框,或者说是保留这些边界框中最好的一个。如果对IOU等知识不了解的可以看我上篇博客Pytorch机器学习(五)——目标检测中的损失函数(l2,IOU,G