- 博客(13)
- 收藏
- 关注
RSS订阅原创 Tensorrt安装的注意事项
这个文件夹的命名会提示你，此版本的tensorrt所对应的最高版本的cuda和最高版本的cudnn，你在安装的时候要注意版本，否则会出现奇奇怪怪的报错。建议安装cuda的时候可以先参考一下pytorch官网上常见版本所对应的cuda版本，比如11.3和11.6。下载tensorrt的时候右侧能够看到当前你需要下载的tensorrt所适配的最低版本的cuda，你只需要后期根据上面的版本号的提示来进行下载cuda即可。
2023-08-18 16:51:06
151
1
原创 常见的损失函数在pytorch框架的应用（持续更新）
这里写目录标题前言二分类损失函数多分类损失函数前言目前在比较流行的深度框架tensorflow、pytorch中损失函数都已经被封装的很好了，大家在使用的过程中直接调用就可以了，但是时间长了，大家就可能不记得函数内部的数学逻辑了，在这里写这一片文章的主要的原因就是方便忘记的时候来进行回忆。二分类损失函数二分类损失函数主要是用在分类任务中，对应的激活函数是sigmoid函数，下面我们来进行展开二分类损失函数的公式、意义和与之对应的sigmoid函数。以神经元来进行举例子，通常一个神经元的输出表达形
2022-03-21 15:12:48
3012
原创 torchvision将cv.imread读进来的图片转换成Tensor：transforms.ToTensor()
1、转变形状，将H,W,C转变成C,H,W，这一点可以用在pytorch的模型训练过程中。2、转变数据格式，该API返回的数据是tensor格式的。3、改变数据的范围，可以将数据的范围变成[0,1.0]。
2021-10-18 14:42:53
1782
1
原创 os.path.join()函数用法重点详解
1、不加符号，会返回加完符号的结果os.path.join('path','abc','yyy')>>>'path\\abc\\yyy'2、加符号，从最后一个以”/”开头的参数开始拼接，之前的参数全部丢弃。os.path.join('/aaa','/bbb','/ccc.txt')>>>'/ccc.txt'3、同时存在以‘’./’与‘’/’’开始的参数，以‘’/’为主，从最后一个以”/”开头的参数开始拼接，之前的参数全部丢弃。>>> p
2021-10-17 20:40:46
1555
原创 random.Random().shuffle()的使用方法
上面的只要是后面的Random的数字是一样的这样生成的数字就是一样的，这样就可以类比打乱的API了，如果在模型训练数据生成的过程中数据集中出现random.Random(1337).shuffle(img_paths)，目标集中出现 random.Random(1337).shuffle(mask_paths)，那么生成之后的两个列表也是一一对应的。...
2021-09-24 15:49:39
675
原创 pytorch中的permute，transpose和view之间的区别。
1、permute，具体的例子如下：import torchgrid_x = torch.arange(5).repeat((5, 1)).view((1, 1, 5, 5)).float()# 格式：1,1,5,5grid_y = grid_x.permute(0, 1, 3, 2)print(grid_x)print(grid_y)结果如下：tensor([[[[0., 1., 2., 3., 4.], [0., 1., 2., 3., 4.],
2021-08-13 13:37:49
658
1
原创 torch.stack和torch.cat在深度学习中的用法
1、关于torch.stack做拼接的时候，无论是按照那个维度进行拼接，其结果都会进行扩维的，下面来看具体的例子：import torcha=torch.tensor([[1,2,3],[4,5,6]])b=torch.tensor([[7,8,9],[10,11,12]])c=torch.stack((a,b), 2)print(c)d=torch.stack((a,b),1)print(d)e=torch.stack((a,b),0)print(e)结果如下：tensor([[[
2021-08-13 08:50:36
599
原创 腾讯云服务器小白重装系统
腾讯云服务器重装系统1、百度搜索腾讯云服务打开首页右上角的控制台点击轻量化服务器点击更多中的管理然后找到更多信息中的重置应用然后选择镜像系统删改你操作全部进行完毕之后，就可以耐心等待几十秒就可以了。有时重装之后在Finalshell出现登陆不上的现象，此时直接将服务器的密码在腾讯客户端设置一下，然后直接在Finalshell重新输入密码重新登陆一下就可以了。...
2021-06-27 09:55:04
403
原创 tensorflow创建模型时，在训练过程中对模型参数的一种保存方式。
1、首先确定模型参数要保存的位置 checkpoint_dir = './training_checkpoints'
2021-06-18 21:25:06
958
原创 NMS（非极大值抑制）
非极大值抑制（Non-Maximum Suppression，NMS），顾名思义就是抑制不是极大值的元素。例如在行人检测中，滑动窗口经提取特征，经分类器分类识别后，每个窗口都会得到一个分数。但是滑动窗口会导致很多窗口与其他窗口存在包含或者大部分交叉的情况。这时就需要用到NMS来选取那些邻域里分数最高（是行人的概率最大），并且抑制那些分数低的窗口。 NMS在计算机视觉领域有着非常重要的应用，如视频目标跟踪、数据挖掘、3D重建、目标识别以及纹理分析等 。...
2021-05-28 20:19:43
689
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人