Pytorch
文章平均质量分 70
烤粽子
目前专注3D目标检测,自动驾驶。
展开
-
mmdetection/mmdetection3d多机多卡训练
因为3d检测训练时间太久,所以想要在mmdet3d上开多机,发现加载完标注文件pkl/json之后,卡住了,找到如下报错。其中有个warning :using best-guess GPU, 大概率是rank不对,没什么问题,按照提示修改。...原创 2022-08-02 20:17:29 · 2884 阅读 · 0 评论 -
pytorch DataLoader(2): Dataset,DataLoader自定义训练数据_opencv,skimage,PIL接口
本文主要简单讲解一下opencv,skimage,PIL读取数据并加载到dataset。一些前置知识可以参考前一篇文章pytorch DataLoader(1): opencv,skimage,PIL,Tensor转换以及transforms,这篇文章主要讲了一些opencv,skimage,PIL的格式,读取方式,相互转换等,有助于帮助大家理解本文本文的一些操作等。import osimport numpy as npimport cv2import torchfrom torch.utils原创 2021-05-25 18:41:48 · 1684 阅读 · 4 评论 -
pytorch DataLoader(1): opencv,skimage,PIL,Tensor转换以及transforms
前置知识在使用pytorch进行dataload,transform之前,需要了解一些数据的知识,许多人使用不同的接口因为不熟悉犯了一些错误。在这里对一些常用的OpenCV,PIL,skimage进行了一些总结,以及pytorchvision.transorforms的一些简单使用。import cv2from PIL import Imagefrom skimage import io, transform, colorimport matplotlib.pyplot as pltimport原创 2021-05-24 18:08:44 · 4317 阅读 · 18 评论 -
将Albumentations用于语义分割任务
将Albumentations用于语义分割任务本文主要基于Using Albumentations for a semantic segmentation task的翻译修改,有少量修改,添加了一些个人理解批注。Colab notebook示例GitHub notebook示例一些安装和数据获取的步骤,在此就不再赘述了。import randomimport cv2import torchfrom matplotlib import pyplot as pltimport albume翻译 2021-05-28 14:41:14 · 1828 阅读 · 0 评论 -
图解pytorch padding方法 ReflectionPad2d
torch.nn.ReflectionPad2d(padding)这个函数简单来说就是:利用输入边界的反射来填充输入张量。官方文档里给了该padding的输入输出如下所示:CLASS torch.nn.ReflectionPad2d(padding: Union[T, Tuple[T, T, T, T]])Input: (N,C,Hin,Win)(N, C, H_{in}, W_{in})(N,C,Hin,Win)Output: (N,C,Hout,Wout)(N, C, H_{ou原创 2021-02-22 00:11:06 · 2876 阅读 · 2 评论 -
Pytorch多GPU训练
Pytorch多GPU训练1. torch.nn.DataParalleltorch.nn.DataParallel()这个主要适用于单机多卡。个人一般比较喜欢在程序开始前,import包之后使用os.environ['CUDA_VISIBLE_DEVICES']来优先设定好GPU。例如要使用物理上第0,3号GPU只要在程序中设定如下:os.environ['CUDA_VISIBLE_DEVICES'] = '0,3'**注意:**如上限定物理GPU后,程序实际上的编号默认为device_ids[原创 2021-01-22 17:42:47 · 2054 阅读 · 0 评论 -
Pytorch CUDA报错,RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
(多卡环境)在使用torchsummary()进行可视化的时候,代码报错:RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR在代码中设置指定GPU,代码仍旧运行不成功device = torch.device(“cuda:3” if torch.cuda.is_available() else “cpu”)model = model.to(device)import torchfrom torchsummary import su原创 2021-01-11 16:33:21 · 22287 阅读 · 3 评论 -
torch.split()
torch.split(tensor, split_size_or_sections, dim=0)Splits the tensor into chunks. Each chunk is a view of the original tensor.tensor (Tensor) – tensor to split. 输入split_size_or_sections (int) or (list(int)) – size of a single chunk or list of sizes f原创 2021-01-08 14:09:59 · 500 阅读 · 0 评论