![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
蒽,开心(∩_∩)
这个作者很懒,什么都没留下…
展开
-
cuda找不到路径报错
编译C++文件时出现:error: [Errno 2] No such file or directory: ':/usr/local/cuda:/usr/local/cuda/bin/nvcc。在终端输入: export CUDA_HOME=/usr/local/cuda。原创 2023-03-24 11:01:52 · 1242 阅读 · 0 评论 -
cuda找不到路径报错
编译C++文件时出现:error: [Errno 2] No such file or directory: ':/usr/local/cuda:/usr/local/cuda/bin/nvcc。在终端输入: export CUDA_HOME=/usr/local/cuda。原创 2023-03-01 23:06:11 · 1066 阅读 · 0 评论 -
目标检测之AP、mAP计算
最近在做模型评测,在评测指标的计算过程上卡了很久,之前在课本上看的计算过程以及很多博客上看计算方法,介绍得比较简单,真正自己算的时候才发现细节真的实在太多了,花了一天时间理清了计算过程。1.目标检测结果输出与指标计算流程(1)以基于anchor的目标检测为例,模型输出大量检测框的类别分数和对应的offset,offset解码得到预测框的左上角和右下角坐标:(x1,y1,x2,y2);(2)解码之后得到的检测框数量是远远大于真实数量的,存在多个预测框重叠的情况,于是使用NMS的方法对预测框进行过滤,注原创 2022-05-22 11:05:29 · 570 阅读 · 0 评论 -
win10系统目标检测、图像识别环境搭建踩坑笔记
1.pycocotools安装失败直接从这个网站下载pycocotools安装包,下载自己需要的版本,然后在anaconda prompt里面cd到该文件的路径,输入pip install pycocotools_windows-2.0.0.2-cp38-cp38-win_amd64.whl,成功安装https://pypi.org/project/pycocotools-windows/#files2.from torch.utils.tensorboard import SummaryWrite原创 2022-03-29 23:29:16 · 581 阅读 · 0 评论 -
torch使用tensorboard可视化训练过程
1.导入模块from torch.utils.tensorboard import SummaryWriter2.在训练开始之前定义一个writer,括号内为tensorboard保存的文件夹地址:writer = SummaryWriter('./runs')3.训练过程中每个epoch写入相应的点:writer.add_scalar('LOSS/Train_loss', float(avg_train_loss),(epoch + 1))writer.add_scalar('LOSS/V原创 2022-03-11 17:52:23 · 1054 阅读 · 0 评论 -
ImageFolder数据读取的标签对应关系
pytorch有自带的库,torchvision.datasets.ImageFolder用于读取数据集,将数据集放在相应文件位置下,文件名就是标签,效果如下:数据读取方式:train_datasets = datasets.ImageFolder(train_path, transform=train_transforms)train_data_size = len(train_datasets)train_data = torch.utils.data.DataLoader(train_dat原创 2022-03-11 13:12:17 · 4822 阅读 · 0 评论 -
torch.argmax()与torch.max()的使用方法及区别
1.torch.max()分类神经网络的输出是所有类别对应的概率值,要返回标签的话就需要用到将概率值与标签对应。torch.max()返回tensor数据最大值和索引,输出的值有两个参数,第一个参数是最大值,第二个参数是最大值的索引(也就是分类label),主要用于神经网络输出与label的匹配。代码如下:import torchoutput=torch.tensor([[-0.7403, -0.6481], [-0.6869, -0.6994], [-0.6569,原创 2022-03-11 11:17:23 · 4798 阅读 · 0 评论 -
Ubuntu、win10系统pytorch cuda版本安装
深度学习常用pytorch框架,安装之前需要确保已经配置好显卡驱动和CUDA,具体教程后面再补充。1.torch安装:进入pytorch官网,选择对应版本进行安装,复制粘贴红框中的命令,进入prompt安装就可以了,效果如下:2.安装验证:验证安装版本和GPU是否可用:import torchprint(torch.__version__)print(torch.cuda.is_available())out:1.10.2True...原创 2022-03-05 11:26:01 · 432 阅读 · 0 评论 -
map()函数给对dataframe数据替换标签
神经网络label一般为str数据类型,需转为结构数字输入,pytorch自带的datasets.ImageFolder库使用文件夹名称自动生成,针对dataset和label分别在两个文件夹的数据集,可将数据和标签保存至CSV文件,使用map()函数统一将字符串标签改为数字。代码:import pandas as pddf=pd.DataFrame()df['imges']=['weather','timeof day','scene','roadtype','weather','weather'原创 2022-02-24 14:48:49 · 2051 阅读 · 0 评论 -
pytorch导入自定义数据集
最近刚学图神经网络,数据集导入折腾了很久,终于开窍了一点。目前常用的数据导入方法主要有两种:(1)torchvision自带的导入方式:这种导入方式使用了torchvision自带的库,打开函数进去看它的说明是这样的:直接翻译过来意思就是图片要放在相应类别的文件夹下,文件夹名字就是图片所属的类别。导入代码如下:from torchvision import datasets'''transform可自行定义'''train_transforms = transforms.Compose(原创 2022-02-24 20:48:13 · 5139 阅读 · 0 评论