pytorch修炼之路
文章平均质量分 65
记录在使用pytorch的遇到的一些问题,以及使用pytorch来实现的一些功能
修炼之路
主要研究领域包括图像分类、目标检测、OCR、人脸识别等,搞过跨平台的深度学习模型的部署解决方案,设计过分布式的深度学习模型服务架构。
展开
-
Python解析VOC数据标签文件
在目标检测数据集中我们经常会使用到VOC格式的数据,这篇文章就向大家介绍一下如何来解析,xml库提供了几种不同的方法来解析xml文件原创 2022-07-31 21:20:45 · 1849 阅读 · 0 评论 -
torch里面的Tensor、as_tensor、tensor以及from_numpy究竟有何区别?
导读在使用pytorch的时候我们经常会用到将numpy array转为tensor,以及将tensor转为numpy array的情况。pytroch内置了几种不同的方法可以方便我们将numpy array转换称为tensor,那么这些方法之间究竟有何异同呢?下面我们就来介绍一下实践import torchimport numpy as np#创建一个numpy array的数组array = np.array([1,2,3,4])#将numpy array转换为torch tensor原创 2021-07-05 18:51:24 · 5847 阅读 · 3 评论 -
error: identifier “AT_CHECK“ is undefined
错误原因在使用pytorch编译maskrcnn-benchmark的时候报,如下错误idia/.local/lib/python3.6/site-packages/torch/include/THC -I/usr/local/cuda/include -I/usr/include/python3.6m -c /home/nvidia/maskrcnn-benchmark/maskrcnn_benchmark/csrc/cuda/deform_pool_cuda.cu -o build/temp.li原创 2021-01-28 13:58:17 · 4274 阅读 · 2 评论 -
常用的图像标注工具汇总
导读有时候我们想训练自己的模型比如目标检测、图像分割、文字识别等,而此时开源的数据并不能满足我们的需求,我们就需要自己准备数据了。这里给大家介绍一些常用的图像标注工具,基本上都能满足大家的需求labelImglabelImg是一个图像标注工具,它是使用python的qt开发的。通过它标注图像生成的标签文件支持xml、PASCAL VOC、YOLO。优点:跨平台,支持Linux、Mac OS、Windows安装方便使用简单缺点:只支持矩形框的标注开源地址:https://github.com/原创 2021-01-20 23:14:19 · 28812 阅读 · 2 评论 -
通俗易懂的Spatial Transformer Networks(STN)(二)
导读上一篇通俗易懂的Spatial Transformer Networks(STN)(一)中,我们详细介绍了STN中会使用到的几个模块,并且用pytorch和numpy来实现了他们,这篇文章我们将会利用pytorch来实现一个MNIST的手写数字识别并且将STN模块插入到CNN中STN关键点解读STN有一个最大的特点就是STN模块能够很容易的嵌入到CNN中,只需要进行非常小的修改即可。上一篇文章我们也说了STN拥有平移、旋转、剪切、缩放等不变性,而这一特点主要是依赖θ\thetaθ参数来实现的。刚开原创 2021-01-17 22:52:54 · 4320 阅读 · 3 评论 -
pytorch输出网络每层的参数和维度信息
导读tensorflow提供了一个summary函数可以让我们很方便的查看到网络每层的输出的shape信息和参数信息,pytorch中也提供了类似的方法pytorch输出网络每层信息安装torchsummarypip install torchsummary输出网络信息summary函数介绍model:网络模型input_size:网络输入图片的shape,这里不用加batch_size进去batch_size:batch_size参数,默认是-1device:在GPU还是C原创 2021-01-16 13:01:30 · 17285 阅读 · 3 评论 -
通俗易懂的Spatial Transformer Networks(STN)(一)
导读pytorch为了方便实现STN,里面封装了affine_grid和grid_sample两个高级API。对STN不太了解的同学可以参考这篇详细解读Spatial Transformer Networks(STN)其实STN的作用是想让CNN具备平移、旋转、缩放、剪切不变性,虽然说CNN中的Pooling可以让网络具备一点平移不变性,但这毕竟是隐性的,如果能让网络直接具备这样的能力岂不是更好。如果对图像处理有了解的同学也许听过仿射变换这个名词,我们只需要通过变换矩阵θ\thetaθ(由6个参数组成原创 2021-01-14 20:39:46 · 5336 阅读 · 3 评论 -
RuntimeError: cuda runtime error (98) : unrecognized error code
导读在使用mmdetection训练目标检测模型的时候,报RuntimeError: cuda runtime error (98) : unrecognized error code at mmdet/ops/sigmoid_focal_loss/src/sigmoid_focal_loss_cuda.cu:139环境信息Ubuntu16.04pytorch:1.3.1torhvision:0.4.2mmdet:1.0rc1+173ef21mmcv:0.2.14cuda:10.0原创 2020-12-11 15:15:15 · 2340 阅读 · 0 评论 -
pytorch报indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead
导读在使用pytorch训练的时候报一堆的UserWarning,报错信息如下/pytorch/aten/src/ATen/native/IndexingUtils.h:20: UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead./pytorch/aten/src/ATen/native/IndexingUtils.h:20: UserWarning原创 2020-12-11 15:04:00 · 2879 阅读 · 1 评论 -
conda安装pytorch下载太慢
导读在使用conda安装pytorch的时候下载特别慢,有时候下载到一半了还会断掉,然后还需要重新下载才行,很难受。比如我们要安装pytorch1.3.1的版本,安装命令如下conda install pytorch=1.3.1 torchvision=0.4.2 -c pytorch解决办法设置清华的conda源# ubuntuconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/f原创 2020-12-10 13:48:53 · 5065 阅读 · 1 评论