草庐IT

torch-geometric

全部标签

torch unsqueeze()详解

 Torch官网解释:torch.unsqueeze(input, dim) → Tensor        Returnsanewtensorwithadimensionofsizeoneinsertedatthespecifiedposition.        Thereturnedtensorsharesthesameunderlyingdatawiththistensor函数的功能是将输入增加一个维度,dim决定在哪一层增加一个维度>>>x=torch.tensor([1,2,3,4])>>>torch.unsqueeze(x,0)tensor([[1,2,3,4]])>>>torc

安装torch-scatter/torch-sparse无法继续的解决方法

问题描述使用以下命令安装torch-scatter/torch-sparsepipinstalltorch-scatterpipinstalltorch-sparse但是在执行过程中出现安装包卡住不动,无法继续安装成功的问题,报错如下Buildingwheelfortorch-scatter(setup.py)...原因造成这个错误的原因是相关的wheel文件下载不到或者是下载缓慢解决方法1.使用以下命令查看已经安装的torch的版本piplist结果如下所示,我的版本是1.10.0+cu1132.使用以下命令查看已经安装的python的版本python--version结果如下所示,我的版本

Neural Geometric Level of Detail: Real-time Rendering with Implicit 3D Shapes 论文笔记&环境配置

论文基本信息:发布于CVPR2021创新点论文介绍了一种具有神经SDF的复杂几何实时渲染方法。论文提出了一种神经SDF表示,可以有效地捕获多个LOD,并以最先进的质量重建3D几何图形。论文中的架构可以以比传统方法具有更高视觉保真度的压缩格式表示3D形状,并且即使在单个学习示例中也能跨不同几何图形进行泛化。背景:直接渲染神经sdf,可以使用寻根算法(如球面追踪),进行光线跟踪。Pipeline:SDF的表现形式:d=f(x)是点x到体积M的表面S的最短符号距离,符号表示的x在M的内部或外部。使用与标准的SDF类似,使用神经网络的参数和编码形状的附加学习输入特征来表示SDF。(使用包含特征向量集合

[解决torch.nn.CrossEntropy()] Assertion `t >= 0 && t < n_classes` failed.的一种情况

 错误信息如下../aten/src/ATen/native/cuda/NLLLoss2d.cu:93:nll_loss2d_forward_kernel:block:[0,0,0],thread:[xxx,0,0]Assertion`t>=0&&t运行背景:Linux/Windows都有,我是先用nnUnet预处理了医学数据,由于它有b好多不同的类别,也就是多数据集联合训练,有A,B,C三种类别(A不等于B不等于 C),直接读取NifTi是没有问题的,读取预处理的npz格式数据,开始报错,本来以为是不是类别传错了,缩减到一个数据集也是报错.最后注释掉CrossEntropy()就不报错了,找

anaconda 安装matplotlib 、PIL 、opencv、numpy、mediapipe、skimage、torch torchvision torchaudio、imutils、onnx

anaconda安装包下载包【安装matplotlib】pipinstall-ihttps://pypi.tuna.tsinghua.edu.cn/simplematplotlib==3.5.2【安装PIL】(如果已经安装了matplotlib,则无需单独安装)pipinstallpillow【安装opencv】pipinstall-ihttps://mirrors.aliyun.com/pypi/simple/opencv-python==3.4.11.45pipinstall-ihttps://mirrors.aliyun.com/pypi/simple/opencv-contrib-py

【Ubuntu20.04 CUDA11.1+Torch1.10+Anaconda 保姆级安装教程】

安装CUDA时需要和Torch版本对应起来,最好先去torch官网上确定要安装的torch版本对应的CUDA版本。在安装CUDA之前需要先确定是否已经安装驱动,打开终端输入nvidia-smi,若有输出,则表明驱动安装过,否则需要先安装驱动(驱动安装教程)一、CUDA11.1安装1、CUDA11.1下载先去CUDA官网上下载要安装的版本(CUDA11.1下载链接),依次选择Linux——》x86_64——》Ubuntu——》20.04——》runfile(local),根据自己的电脑配置选择即可打开终端,先复制第一条语句到终端下载CUDAwgethttps://developer.downlo

Anaconda Prompt里面安装了torch,但Pycharm无法成功import torch

许多初学者在安装torch的包时,会遇到这样一个问题,AnacondaPrompt里面安装了torch,但在Pycharm中无法使用,主要是Python解释器没有配置正确。1、查看环境condaenvlist 可以看到有三个,当前是在(base)环境下2、查找torch包piplist 在(base)环境下可以看到已经安装成功了torch包,如果当前环境没有则切换环境查看,都没有的话可能是包没有下载成功,安装包过程这里不再赘叙,可以看下面这条详细版。 https://blog.csdn.net/qq_45281807/article/details/112442423?spm=1001.201

【torch.nn.init】初始化参数方法解读

 可参考:torch.nn.init-云+社区-腾讯云一. torch.nn.init.constant_(tensor, val)  1.作用:    常数分布: 用值val填充向量。2.参数:tensor –ann-dimensionaltorch.Tensorval –thevaluetofillthetensorwith3.实例:importtorchformtorchfromnnw=torch.empty(3,5)print(w)print(nn.init.constant_(w,0.3))-------------------------------------tensor([[6

torch.nn.functional.interpolate()函数详解

    通常可以使用pytorch中的torch.nn.functional.interpolate()实现插值和上采样。上采样,在深度学习框架中,可以简单理解为任何可以让你的图像变成更高分辨率的技术。input(Tensor):输入张量size(int orTuple[int]orTuple[int,int]orTuple[int,int,int]):输出大小scale_factor(floatorTuple[float]): 指定输出为输入的多少倍数。如果输入为tuple,其也要制定为tuple类型mode(str): 可使用的上采样算法,有’nearest’,‘linear’,‘bili

【Python】torch.exp()和 torch.sigmoid()函数详解和示例

本文对torch.exp()和torch.sigmoid()函数进行原理和示例讲解,以帮助大家理解和使用。目录torch.exp函数原理运行示例torch.sigmoid()函数原理运行示例torch.sigmoid相关知识结合运行torch.exp函数原理torch.exp是PyTorch库中的一个函数,它的作用是对输入的张量进行按元素指数运算。指数运算是将基数的指数次幂的结果,例如,计算e的x次幂,其中e是自然对数的底数。torch.exp返回一个新的张量,张量的形状和输入张量相同,但是每个元素都是输入元素的指数运算结果1。函数原型:torch.exp(input)参数:input:输入张