- 博客(19)
- 收藏
- 关注
RSS订阅原创 RuntimeError: Failed to export an ONNX attribute ‘onnx::Cast‘
RuntimeError: Failed to export an ONNX attribute 'onnx::Cast'
2022-08-03 10:28:08
417
原创 python cv2保存图像失败
失败案例： cv2.imwrite(’test1.jpg‘, simi)改为： cv2.imwrite(“test1.jpg”, simi)
2021-11-12 15:38:53
1078
原创 Python 函数参数为列表list或者字典dict，避免修改原值。
参数为list：def fun0(n): n = n * 1 passdef fun1(n): n = n.copy() passdef fun2(n): passa = [1,2,3,4]fun0(a)fun1(a)fun2(a[:])参数为dict：使用fun1的方法
2021-09-17 19:13:07
398
原创 linux 修改默认端口(训练中端口冲突)
linux 修改默认端口(训练中端口冲突)在命令前加 PORT=29501 ；例如 CUDA_VISIBLE_DEVICES=0,1,2,3 ./tools/train.sh改为：CUDA_VISIBLE_DEVICES=0,1,2,3 PORT=29501 ./tools/train.sh
2021-03-08 20:58:40
155
原创 进程已死，显存还在被占用(无fuser命令)
由于某种原因，进程挂掉了，但占用的显存未释放。情况：服务器内无fuser，7卡被占用，nvidia-smi显示无对应进程。网上的方法大多都是采用fuser查看，但服务器上没有fuser，也没有权限，网上其他方法也不work，然后自己试出来这个方法。解决：显示所有进程( ps aux|grep python )，找到占用显存的进程并杀死( kill -9 PID )。注：如果不知道是哪些进程占用，可以计算下对应显存大小。如果是自己引起的，直接把自己名下的进程
2020-11-03 16:12:45
2076
1
原创 RuntimeError: CUDA error: device-side assert triggered
RuntimeError: CUDA error: device-side assert triggered 断言形式和下述类似::operator()(int)->auto: block: [0,0,0], thread: [4,0,0] Assertion index >= -sizes[i] && index < sizes[i] && “index out of bounds” failed./opt/conda/conda-bld/pytorc
2020-11-03 08:47:18
909
原创 pytorch 模型参数(卷积)转不到cuda中，RuntimeError: Input type (torch.cuda.FloatTensor) and weight type
pytorch 模型参数转不到cuda中报错：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the错误示例：class my_layer(nn.Module): def __init__(self, in_channels, out_channels, num_layer): super(my_layer, self).__init_
2020-09-11 15:40:21
867
2
原创 pytorch load模型参数(.pth)，RuntimeError: version_ ＜= kMaxSupportedFileFormatVersion INTERNAL ASSERT FAIL
pytorch load模型参数(.pth)，RuntimeError: version_ <= kMaxSupportedFileFormatVersion INTERNAL ASSERT FAIL原因：使用较高版本的pytorch训练得到的模型，在低版本的ptorch中load时，存在的版本问题。我使用1.6.0版本训练的模型，在1.4.0中是load不了的。解决：升级版本或者使用低版本训练。要不找个一种兼容的保持和读取方法。...
2020-08-13 11:41:44
10924
4
原创 在服务器linux中切换nvcc版本
在服务器linux中切换nvcc版本在nvidia-smi 中cuda版本在conda中nvcc版本这里显示当前使用的nvcc版本由于conda中nvcc版本低于要求，所以要更换。直接去/usr/local 中看服务器安装的cuda版本，然后使用10.1，通过命令行vim ~/.bashrc修改conda的cuda版本然后sourcesource ~/.bashrc结果：...
2020-07-29 10:43:31
3614
3
原创 python 对相同数据进行相同计算，结果不同
python 对相同数据进行相同计算，结果不同这种情况现发现在CPU中存在，在GPU中不存在。暂认为是处理时采用的数据精度不同
2020-07-22 21:08:33
1428
原创 Pytorch在forward中创建了新的tensor时，RuntimeError: arguments are located on different GPUs
Pytorch在forward中创建了新的tensor时，RuntimeError: arguments are located on different GPUs这个问题浪费了我一天时间，找了很多关于相关错误提示的解决方案。很不幸，有用但解决不了。下面进入正题：我的错误引起是：我在自定义forward中声明了一个tensorclass test(nn.Module): def __init__(self, xxxx): super(test, self).__init__
2020-07-17 17:07:20
772
原创 pytorch自定义操作层，正向和反向
torch 自定义操作层， 前向和反向过程在torch中定义正向一定要注意是否使用Variable比如，在定义层的forward中正确定义是这样的： def forward(self, infeature): infeature = infeature.reshape(self.inshape[1], self.inshape[2]) / self.p_pixel_num infeature = torch.mm(infeature, self.pooling)
2020-06-30 09:39:56
699
原创 python 数据中存在nan
python 数据中显示nannan的表示这个数是个非数，就像 1除0.所以检查下哪里存在除0的情况，或者有其他的位置存在计算成非数。nan会随着计算传播至后续计算
2020-06-29 17:21:56
2727
原创 解决AttributeError
解决AttributeError: object has no attribute ‘log_softmax’错误例子：net = FCN(data)直接在构建网络时，给与输入改为：net = FCN()out = net(data)
2020-06-16 19:42:51
2250
1
原创 ONNX生成模型遇到的问题
在使用onnx生成模型遇到的问题onnx.onnx_cpp2py_export.checker.ValidationError: Nodes in a graph must be topologically sorted, however input ‘ends’ of node:input: “x3” input: “starts” input: “ends” output: “Y” op_...
2019-11-06 22:25:40
6557
5
原创 图像贴图（旋转缩放平移）
自己写的一个贴图算法，虽然很简单，很容易就实现了，但还是要记录一下其中过程。其中有个矩阵运算，其实是这样写的。x2 = (int)(exa_x1cos(rot) - exa_y1sin(rot)) ;//旋转后坐标y2 = (int)(exa_y1cos(rot) + exa_x1sin(rot)) ；身为小白，自己实现了这个简单的算法，也是很高兴的。希望对其他人也有所帮助。rz和cz...
2019-07-19 11:01:56
416
原创 基于皮肤分割的磨皮算法
查阅了博客上绝大多数关于皮肤分割和磨皮美颜的算法，然后自己找了各位博主的代码，有自己将各部分代码组合修改调试，最终写了份参照代码。至于引用哪些博主的代码，找的博客太多分不清楚了，有博主看到引用代码可以私聊我加上你的引用。 我感觉做出来的效果不太好，不如别人算法效果好。 此代码只是做了皮肤分割和滤波磨皮，美白的代码还没写。还有就是发布文章我不太懂它的分类选项，只好随便选了。发布文章的规则我也不...
2019-07-16 16:35:20
797
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人