PyTorch
文章平均质量分 71
PyTorch与深度学习
冰激凌啊
下雨打伞
展开
-
深度学习标注结果可视化
对于VMRD数据集:https://gr.xjtu.edu.cn/zh/web/zeuslan/dataset,其中:目标识别是xml格式标注;抓取位置检测是txt格式标注;xml格式标注结果可视化import xml.etree.ElementTree as ETimport os, cv2xml_file = 'C:/Users/00/Desktop/VMRD V2 fixed/Annotations/00077.xml'tree = ET.parse(xml_file)root原创 2022-03-28 17:32:04 · 4102 阅读 · 2 评论 -
将pytorch和ros同时使用——将深度学习的结果发布成rostopic
将pytorch深度学习的结果发布ros话题,只需要注意环境配置的部分即可原创 2021-12-14 17:22:13 · 6713 阅读 · 15 评论 -
Ubuntu下使用yolov5
Ubuntu使用yolov5进行测试,并使用realsense d435i测试原创 2021-12-09 22:32:09 · 7790 阅读 · 4 评论 -
Ubuntu18配置与ROS兼容的深度学习环境(Anaconda3+PyTorch1.10+python3.8+cuda10.2)
ubuntu18同时安装anaconda和ros,保证二者不冲突,并创建pytorch环境原创 2021-12-06 15:43:52 · 5741 阅读 · 8 评论 -
使用自己的数据集,修改运行faster RCNN
找了一个faster rcnn的demo,但是因为初学,即使有一个例程也不太会使用自己的数据集使它正常运行。现记录一下针对自己数据集的修改过程:01 标注数据集第一步:将标注信息放入MyAnnotations标注数据集,由于之前的demo,数据集都是放入VOCdevkit文件夹,方便起见,我们也放入VOC2012文件夹下。同时,为了和PASCAL VOC2012数据集做区分,我们创建MyAnnotation和MyJPEGImage两个文件夹。前者放入我们自己数据集的标注信息,后者放入JPEG格式的原创 2021-10-19 11:43:47 · 2529 阅读 · 2 评论 -
pytorch的主要模块
01 torch模块激活函数Sigmoid (torch.sigmoid)ReLU (torch.relu)Tanh (torch.tanh)张量矩阵的乘法 (torch.mm)张量元素的选择 (torch.select)能够产生一定形状的张量torch.zeros产生元素全为0的张量torch.randn产生元素服从标准正态分布的张量02 torch.Tensor模块torch.Tensor模块定义了torch中的张量类型,其中的张量有不同的数值类型,如单精度、双精原创 2021-09-01 22:15:08 · 899 阅读 · 0 评论 -
PyTorch官方demo:图像分类 LeNet
是参考B站一位up主的,感觉讲的很好,讲解的每节课程在csdn上做了一个目录:https://blog.csdn.net/qq_37541097/article/details/103482003这一节的up主的参考资料如下:b站讲解视频:https://www.bilibili.com/video/BV187411T7Ye代码:https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/tree/master/pytor原创 2021-09-05 21:56:19 · 448 阅读 · 0 评论 -
win10配置深度学习环境(anaconda3+pytorch1.9+python3.7+cuda10.2)
1 下载VS2019安装地址:https://visualstudio.microsoft.com/zh-hans/vs/安装参考:Microsoft Visual Studio VS2019中文版本软件下载及安装教程!Win10 系统在 Visual Studio 2019 环境下配置 CUDA 10.1 + TensorFlow-GPU 1.14.01 下载Anaconda下载地址:https://www.anaconda.com/products/individual原创 2021-08-26 20:31:15 · 1547 阅读 · 1 评论 -
TensorFlow/PyTorch和cuda等版本对应关系
和python的冲突问题https://zhidao.baidu.com/question/1116512378559354699.htmlpython与pytorch、显卡、显卡驱动、cuda版本对应关系https://blog.csdn.net/BigData_Mining/article/details/104983246参考:https://www.jianshu.com/p/ac70300b598bNVIDIA系列显卡做深度学习,需要在主机安装显卡驱动(driver)在doc原创 2021-08-24 11:42:24 · 5722 阅读 · 0 评论 -
ResNet和迁移学习
参考:https://www.bilibili.com/video/BV14E411H7Uw在头文件中导入pytorch官方的resnet网络import torchvision.models.resnet然后打开resnet.py在下述文件中下载网络,在train.py中载入如果不想使用迁移学习,想自己训练,可以将下面标记的这段删去,然后在net = resnet34()中传入参数...原创 2021-09-07 21:52:01 · 436 阅读 · 0 评论 -
用PyTorch构建一个简单线性回归模型
01 PyTorch模块类的构建方法首先导入torch.nn库,然后基于继承nn.Module的方法构建深度学习模块。整个模块的函数主要由两部分组成:通过__init__ 方法初始化整个模型forward方法对该模型进行前向计算。其中,在使用__init__ 方法的时候,可以在类内部初始化子模块,然后在forward方法中调用这些初始化的子模块,最后输出结果张量。可以给类取任意的名字。因为需要调用父类nn.Module的初始化方法,这里需要使用super函数来获取当前类的父类(即nn.Mo原创 2021-09-03 16:49:00 · 454 阅读 · 0 评论 -
CUDA out of memory问题解决
在使用resnet demo训练过程中,使用CPU训练不会报错。但在使用GPU(GTX1650 4GB)训练时提示如下错误:RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 4.00 GiB total capacity; 550.99 MiB already allocated; 1.90 GiB free; 572.00 MiB reserved in total by PyTorch)解决:改小bat原创 2021-09-07 21:28:25 · 2183 阅读 · 0 评论