
pytorch
文章平均质量分 52
Irving.Gao
中国科学技术大学在读研究生,自动驾驶场景理解方向。
研究方向:自动驾驶3D计算机视觉感知任务,图像分割,BEV,目标检测。
本科经历:大创项目-基于物联网的沙漠治理智能机器人集群、智能车、ROBOCON等机器人相关比赛,曾获挑战杯、智能车等国家一等奖四项等数十项国家级奖项,曾在杭州宇树科技、深圳科创学院智能驾驶中心、上海纽励科技、商汤科技实习。
展开
-
Ubuntu18.04——私房整理安装NIVIDA驱动(11.4), CUDA11.2, cudnn8.2, TensorRT8.0, Pytorch1.9.0,PaddlePaddle2.2.2
文章目录安装NVIDIA驱动安装CUDA和cudnncudnn安装测试安装版本信息:NIVIDA Driver:±----------------------------------------------------------------------------+| NVIDIA-SMI 470.94 Driver Version: 470.94 CUDA Version: 11.4 ||-------------------------------±---原创 2022-02-27 21:46:45 · 4231 阅读 · 3 评论 -
Pytorch——基础API用法总结
文章目录TensorPytorch DocumentationTensorTensor.max()B, N, C, H, W = feat.shapefeat = feat.max(dim=1)feat.shape # [B, 1, C, H, W]Tensor.view()B, N, C, H, W = feat.shapefeat = feat.view(B, N, C, H*W)Tensor.permute()B, N, C, H, W = feat.shap原创 2022-05-04 17:32:35 · 2332 阅读 · 1 评论 -
t-SNE——高维特征可视化分析
无监督学习之t-SNEhttps://github.com/SarthakV7/Clustering-Barron-s-333-word-list-using-unsupervised-machine-learning原创 2023-09-13 19:04:58 · 374 阅读 · 0 评论 -
Pytorch——查找、替换module相关操作
【代码】Pytorch——查找、替换module相关操作。原创 2023-09-11 14:38:40 · 966 阅读 · 1 评论 -
MMDet3D——数据增强Pipline‘GlobalRotScaleTrans‘和‘RandomFlip3D‘的Pytorch逆变换实现
在点云的3D感知算法中,常用和的数据增强方式,这两个可以有效地增强模型的鲁棒性,提升模型的性能。而本文的出发点在于当我们同时对相同的data使用了多种不同的随机增强方式时,当我们需要在模型中将不同增强方法下的或者GT 3D Box进行对齐时,就需要涉及到逆变换。尤其适用于半监督等同时有label和unlabel data的代码。简单粗暴,我们直接上代码,注释直接写在代码中:(代码可以直接copy使用)原创 2023-06-05 21:53:56 · 1450 阅读 · 0 评论 -
Pytorch——根据2D bounding box生成点采样空间
请注意,示例代码中的坐标范围是在 [-0.5, 0.5] 内进行归一化处理,以适应等间距的网格。如果你的实际应用需要不同的坐标范围或网格密度,可以相应地进行调整。x7的等间距网格坐标,然后根据每个box的中心坐标和宽高计算采样点的坐标。最后,将采样点的坐标展平成形状为。的张量,并返回该结果。原创 2023-05-23 15:07:06 · 249 阅读 · 0 评论 -
Pytorch——在一个Tensor中随机保留一个True,其他全为False
函数从索引中随机选择一个索引,最后创建一个新的张量,只在随机选择的索引处设置为。运行上述代码将输出类似如下的结果,其中只有一个。这种方法可以并行处理多个布尔类型张量。,从而实现了随机只保留一个。函数获取原始张量中所有。要将一个布尔类型的张量。被保留,其他位置都是。原创 2023-05-22 22:33:46 · 687 阅读 · 0 评论 -
Pytorch——报错解决:多卡训练超时错误Timed out initializing process group in store based barrier on rank
【代码】Pytorch——报错解决:多卡训练超时错误Timed out initializing process group in store based barrier on rank。原创 2023-04-08 13:09:48 · 7018 阅读 · 4 评论 -
Pytorch——报错解决:error in ms_deformable_im2col_cuda: invalid configuration argument
【代码】Pytorch——报错解决:error in ms_deformable_im2col_cuda: invalid configuration argument。原创 2023-03-17 13:40:13 · 2130 阅读 · 1 评论 -
Pytorch——对3D Voxel进行任意尺度放缩
【代码】Pytorch——对3D Voxel进行任意尺度放缩。原创 2023-03-02 15:14:08 · 605 阅读 · 0 评论 -
Pytorch——对2D Image \ 3D Voxel特征指定3D点进行Grid Sample采样
因为在做3D的voxel相关工作,需要对某些3D feat中特征点进行采样,而网上大多文章都没讲清楚,于是在这记录并分享一下自己的理解。原创 2023-03-01 22:11:53 · 1919 阅读 · 0 评论 -
Pytorch——报错解决:‘torchtext.data‘ has no attribute ‘Iterator‘
解决方法将改成:原创 2022-12-05 15:09:34 · 1835 阅读 · 3 评论 -
Pytorch——安装torchtext-cuda
安装对应版本torchtext:即可成功安装~参考文章:原创 2022-12-05 14:58:40 · 1572 阅读 · 0 评论 -
CUDA——编写并调用自定义CUDA Pytorch算子
编写并调用自定义CUDA Pytorch算子原创 2022-11-23 19:40:31 · 1785 阅读 · 0 评论 -
Pytorch——报错解决:No inf checks were recorded for this optimizer.
报错解决:No inf checks were recorded for this optimizer.原创 2022-11-18 17:40:50 · 6342 阅读 · 0 评论 -
Pytorch——基于max pooling实现图像腐蚀和膨胀
【代码】Pytorch——基于max pooling实现图像腐蚀和膨胀。原创 2022-11-15 23:55:27 · 1570 阅读 · 0 评论 -
Pytorch——报错解决:AttributeError: module ‘setuptools._distutils‘ has no attribute ‘version‘
【代码】Pytorch——报错解决:AttributeError: module ‘setuptools._distutils‘ has no attribute ‘version‘原创 2022-11-11 17:01:58 · 3243 阅读 · 0 评论 -
Pytorch(Tensor)-Numpy(ndarrays) API对照表
Numpy与Torch API总结原创 2022-11-05 21:11:43 · 2348 阅读 · 2 评论 -
MMDet——EMA更新hook详解
【代码】MMDet——EMA更新hook详解。原创 2022-11-01 17:08:56 · 1967 阅读 · 0 评论 -
Pytorch——报错解决:TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, found
主要原因是由于获取到batch中输入有None,反推代码,说明你的。因此,解决方法就是不要让返回值有None即可。函数返回值为None,因此出现如下报错。原创 2022-10-31 16:13:10 · 12096 阅读 · 0 评论 -
Pytorch——报错解决:RuntimeError: Output 0 of SelectBackward is a view and is being modified inplace.
但是在多卡分布式过程中,不允许出现同一个model前向两次的情况,会报错(我在做Semi-Supervised过程中,需要分别计算。原创 2022-10-30 22:56:45 · 2075 阅读 · 0 评论 -
Pytorch——报错解析:RuntimeError: Expected to mark a variable ready only once. This error is caused by one
我在做Semi-Supervised任务时遇到了该BUG,因为在写模型时将student的model运行了两次,分别对两组image进行了forward train并计算了loss,因此出现了如下错误,以此记录。原创 2022-10-30 18:08:44 · 8141 阅读 · 29 评论 -
Pytorch——ConcatDataset
【代码】Pytorch——ConcatDataset。原创 2022-10-28 18:34:03 · 3832 阅读 · 0 评论 -
Pytorch——Test Time Augmentation
TTA原创 2022-10-26 17:07:07 · 1084 阅读 · 0 评论 -
Pytorch——常用损失函数详解
深度学习之常用损失函数损失函数loss大总结损失函数(Loss Function)原创 2022-10-09 16:31:40 · 7292 阅读 · 0 评论 -
Pytorch——报错解决:匈牙利匹配
学习率太高,导致不收敛cost为nan。原创 2022-10-06 12:21:55 · 1589 阅读 · 0 评论 -
pytorch——计算两个等大无序点云中的距离Earth Mover Distance
目前要做两个无序点集之间的相似性计算,在看过Chamfer Distance后,个人觉得CD的计算方式决定了其无法处理两个点集整体分布差异大,但是局部有部分点距离很近的情况,而这种情况在点集中是一定可能出现的,因此使用EMD可以保证每个点集都有一一对应的整体cost最小点。原创 2022-10-02 11:38:04 · 1678 阅读 · 1 评论 -
Pytorch——报错解决:Function MmBackward returned an invalid gradient at index 0 - got [50, 256]
我的错误出现在Transformer模块的Decoder中原创 2022-09-28 14:20:11 · 1530 阅读 · 0 评论 -
MMDet——基于Chamfer Distance评估点集
目标检测mAP计算以及coco评价标准图解 Chamfer Distance。原创 2022-09-25 16:13:06 · 683 阅读 · 0 评论 -
Pytorch——基于mmseg/mmdet训练报错:RuntimeError: Expected to have finished reduction in the prior iteration
训练报错解决原创 2022-07-23 01:03:43 · 4771 阅读 · 2 评论 -
Pytorch——报错解决:“torch/optim/adamw.py” beta1, UnboundLocalError: local variable ‘beta1‘
UnboundLocalError原创 2022-06-20 21:17:11 · 778 阅读 · 0 评论 -
Pytorch——实现Tensor矩阵的任意角度旋转、平移操作
1原创 2022-06-17 22:33:50 · 7687 阅读 · 7 评论 -
Pytorch——代码导致的异常报错:RuntimeError: CUDA out of memory.
1原创 2022-06-17 17:16:52 · 795 阅读 · 2 评论 -
Pytorch——distributed单机多卡分布式训练
文章目录运行参考文章:pytorch多gpu并行训练运行python -m torch.distributed.launch --nproc_per_node=NUM_GPUS_YOU_HAVE main.pynproc_per_node: 这个参数是指你使用这台服务器上面的几张显卡原创 2022-05-15 21:37:02 · 590 阅读 · 0 评论