- 博客(21)
- 收藏
- 关注
RSS订阅原创 vscode设置debugconda json文件
注意 ：which python的路径记下来： 用于在launch.json添加在"pythonPath"1 查看conda env环境安装位置和对应python。“name”: “Python 调试程序: 当前文件”,手动添加：“pythonPath”
2024-06-07 13:06:54
134
原创 [pytorch] 模型加载，torch.load , keys not match
可能会遇到有一个多gpu的训练后保存的模型，但是后续要在单gpu的机子上使用，在torxh.load时报错。这是因为nn.DataParallel会在模型参数结构前面加一个module.比如你是这么save的state = {'epoch': epoch, 'state_dict': self.model.state_dict(), 'optimizer': self.optimizer.state_dict(), '
2022-03-26 16:11:17
1908
原创 caffe2线程泄露 torch1.9 独有warning & docker 容器exit code
参考于https://blog.csdn.net/qq_41963227/article/details/119391566记录一下！！caffe2线程泄露问题描述：在使用pytorch1.9的dataloader时，如果设置num_workers>0或者pin_memory=True时，出现caffe2线程泄露的warning,数量等于num_workers的数量。[W pthreadpool-cpp.cc:90] Warning: Leaking Caffe2 thread-pool
2021-12-24 17:23:44
1036
原创 docker环境 下安装minkowskiengine 和open3d
最近用到了Minkowskiengine， 实验版本0.4.3~·个人觉得是最简单的编译方式了，make命令不需要在docker下安装，特此记录1 首先pull镜像， anyway 基于个人实践经验ubuntu18.04比 20.04友好。FROM nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04RUN apt-get update RUN apt-get install -y software-properties-common镜像构建完毕，这里关于镜像创
2021-12-15 13:15:49
2076
原创 loss.backward()报错runtime error: one of the variables needed for gradient computation
1 这是因为 在计算梯度前此变量被修改。解决办法“(1)把激活函数的 inplace= True 全部改为falsetorch.relu(inplace =False)(2) 赋值符 += => +a+=b # 修改为a =a+ b
2021-12-14 17:15:38
2185
原创 docker下open3d安装记录
1 理论上 pip install open3d就行了，但是是在docker下安装很多依赖都是没有的，pip install 后import 会报错，如：…while loading shared libraries: libc++.so.1 cann… 特此记录需要安装依赖：apt-get updateapt-get install libc++1然后缺什么包就再安装什么就ok ，如plyfile。pip3 install plyfile...
2021-12-10 11:51:34
1108
原创 open3d获取点云法向量，并指定方向
官方文档:官方文档一 获取点云：open3d支持ply等形式直接读取就可o3d.io.read_point_cloud，但是不支持.bin文件所以可以numpy读取bin文件再转换成open3d的数据格式转换的代码：points = np.random.uniform(-1, 1, (10000, 3))pointcloud = o3d.geometry.PointCloud()pointcloud.points = o3d.utility.Vector3dVector(points)
2021-12-09 12:27:29
2875
原创 PCT的 part_seg 模型 详细信息 记录
PointTransformerSeg((backbone): Backbone((fc1): Sequential((0): Linear(in_features=19, out_features=32, bias=True)(1): ReLU()(2): Linear(in_features=32, out_features=32, bias=True))(transformer1): TransformerBlock((fc1): Linear(in_features=32, out_
2021-11-18 23:56:31
2307
原创 import cv2的 问题，，docker中遇到，ImportError: libGL.so.1: cannot open shared object file: No such file or d
import cv2的 问题，，docker中遇到，pip install opencv-pytorch 可以，但是 import的时候报错缺少安装依赖：apt install libgl1-mesa-glx执行：实现不行的话就素有的依赖都安装一遍，！！2 或者 在 dockerfile里面加上RUN apt-get update RUN apt-get install ffmpeg libsm6 libxext6 -yRUN apt-get install -y python3-
2021-11-17 14:49:21
673
原创 RuntimeError: CUDA error: device-side assert triggered CUDA kernel
报错：RuntimeError: CUDA error: device-side assert triggered CUDA kernel看起来是 runtimeerror，指向性很模糊，其实是 index溢出的问题，比如你的loss_list,data,或者什么数组的 取值超出范围了，特此记录，找了一天。。。。哈！...
2021-11-17 13:30:59
553
原创 Diceloss
pytorch中的 Diceloss 记录https://github.com/BloodAxe/pytorch-toolbelt/blob/0.4.1/pytorch_toolbelt/losses/dice.py输入：y_pred: NxCxHxW y_true: NxHxWfrom typing import Listimport torchimport torch.nn.functional as Ffrom pytorch_toolbelt.utils.torch
2021-11-14 20:59:40
969
原创 pytorch中get_device， 类型转换
get_device2 numpy的item() 获取对应的值np.random.seed(123)x = np.random.randint(9, size=(3, 3))xarray([[2, 2, 6], [1, 3, 6], [1, 0, 1]])x.item(3)1x.item(7)0x.item((0, 1))2x.item((2, 2))1tensor中的item（），获取值并保留2位小数>>> x = tor
2021-11-11 23:07:50
2181
原创 python安装hydra
1.安装记录安装的pypi名字不是Hydra 而是 如下 特此记录pip install hydra-core然后等待就可以啦
2021-11-11 23:06:57
6966
1
原创 spconv2.1使用记录 .features
spconv2spconv2已经可以直接pip安装了，真的非常友好！最好安装最新版本的pcmm，否则会报错，哈哈哈哈哈感谢作者所提供的使用引导参考参考链接改动1 import spconv需要 import spconv.pytorch as spconv2 .features版本2是可以read的，但是进行修改原来的属性.features需要进行更改为x.features = F.relu(x.features)->x = x.replace_feature(F.relu(x.
2021-11-09 18:11:19
2027
原创 PolarSeg源码解析
模型打印理解my_model：ptBEVnet((PPmodel): Sequential((0): BatchNorm1d(9, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)(1): Linear(in_features=9, out_features=64, bias=True)(2): BatchNorm1d(64, eps=1e-05, momentum=0.1, affine=True, track_ru
2021-10-20 22:34:05
337
原创 python的argparse，logging
学习用到了，在此记录，可选参数，必选参数，action = 'store_true’设置True/False参考手册~import argparseparser = argparse.ArgumentParser()parser.add_argument("square", help="display a square of a given number",type=int) # 必选的parser.add_argument("--batch", help="display a squa.
2021-10-11 20:57:34
197
1
原创 torch_scatter安装步骤
查看自己的torch和cuda或者cpu版本import torchtorch.__version__版本一定需要正确匹配2 ## pip或者conda直接安装不可以自行下载手动安装官网：https://pytorch-geometric.com/whl/https://pytorch-geometric.com/whl/torch-1.9.0%2Bcu111.html3 pip install .whl文件搞定！...
2021-10-04 22:59:12
4686
1
原创 Swin transformer学习记录
Swin transformer学习记录新一代backbone，！！核心为Swin Transformer block：MSA+ SW_MSA实现了类似CNN的pooling效果
2021-09-16 23:10:50
88
原创 FCN 论文代码解读(二)-模型构建
FCN 论文代码解读二–模型构建FCN参数类别数class FCN(nn.Module): def __init__(self, num_classes): super().__init__() self.stage1 = pretrained_net.features[:7] # 一第一个block conv relu bn pooling 一共6层 self.stage2 = pretrained_net.features[7:14]
2021-09-15 20:34:48
258
原创 FCN论文以及源码拆分详解（一）
FCN论文以及源码拆分详解（一）FCN 论文：Fully Convolutional Networks for Semantic Segmentation参考github代码。摘要：开山之作-----state of-the-art segmentationOur key insight is to build “fully convolutional” networks that take input of arbitrary size and produce correspondingly-s
2021-09-15 17:21:47
531
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人