logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

pytorch torchvision.ops.roi_align 示例

import torchvisionimport torchinput_ = torch.rand(3, 128, 24, 96) #[b,c,h,w]bbox = [[0,0.1,0.15,0.4,0.6],[0,0.1,0.15,0.4,0.6],[1,0.01,0.15,0.4,0.6],[1,0.1,0.25,0.7,0.6],[2,0.01,0.14,...

#pytorch#深度学习#python
caffe模型转pytorch---LSTM

之前完成了几个网络的caffe转pytorch。refinenethttps://www.cnblogs.com/yanghailin/p/13096258.htmlrefinedethttps://www.cnblogs.com/yanghailin/p/12965695.html上面那个是提取caffe权重然后转libtorch,下面是直接对应的pytorch版本转li...

#深度学习
目标检测画图显示图

这是我见过最好的代码了,摘录自u版本的yolov3里面的。首先看画出的效果图:每个类用不同颜色框,上面写出类别和分数这些信息,并且是填充。字体大小能够根据图片大小自动调整。def plot_one_box(x, img, color=None, label=None, line_thickness=None):# Plots one bounding box on image i...

#python#深度学习#opencv
crnn pytorch转libtorch转trt问题记录--------libtorch的data_ptr()方法不同版本变化了

大坑!!!首先发现这个问题的时候是在libtorch1.1版本上面没有问题的代码,移植到高版本libtorch1.7,发现同样的代码在高版本上面精度不一样。然后查找原因的时候发现的。运行代码发现没有显存累加情况但是精度不对,不能出效果图。之前的环境虽然存在显存累加问题但是精度是对的可以出效果图。查找问题,先查找trt推理出来的结果发现对不上,再查找输入发现对的上。这里我都是对比的libtor...

#bug#深度学习#opencv +1
crnn pytorch 训练、测试

1.仓库地址https://github.com/meijieru/crnn.pytorch原版用lua实现的:https://github.com/bgshih/crnn需要用到的warp_ctc_pytorch: https://github.com/SeanNaren/warp-ctc2.环境安装普通的环境都可以吧,我是cuda10.0,torch1.2.0python3.6...

#python#深度学习#tensorflow +2
交叉熵 pytorch中的nn.CrossEntropyLoss()函数

参考链接:https://www.cnblogs.com/JeasonIsCoding/p/10171201.htmlhttps://blog.csdn.net/qq_27095227/article/details/103775032二分类的交叉熵公式是:如果是多分类,交叉熵公式如下:这里为了更加直观的理解计算过程,举例说明:比如我们举例:假设我们有个一样本,他经过我...

pytorch 常用函数

通过索引赋值a = torch.zeros([5,5])index = (torch.LongTensor([0,1]),torch.LongTensor([1,2]))a.index_put_((index), torch.Tensor([1,1]))a[index] = torch.Tensor([4,4])print(a)tensor([[0., 4., 0., 0....

caffe模型转pytorch

本文介绍的是直接读取caffe权重,并根据caffe网络一步一步的搭同样的pytorch网络。本文以分割网络segnet为例。首先给出参考链接:https://blog.csdn.net/DumpDoctorWang/article/details/88716962绝大部分就是参考这位博主的。首先是需要配环境,我这配好了segnet caffe的训练环境,还有跑推理的脚本,随便找...

pytorch 孪生网络 训练与预测

# coding: utf-8# # One Shot Learning with Siamese Networks## This is the jupyter notebook that accompanies# ## Imports# All the imports are defined here# In[2]:#get_ipython().run_li...

facebook-pytorch maskrcnn-benchmark训练自己的数据集

数据准备gray标签图转coco数据集 https://www.cnblogs.com/yanghailin/p/11215003.htmlvoc定位数据xml转coco数据集格式json https://www.cnblogs.com/yanghailin/p/11189871.html我把数据准备成coco格式的。在maskrcnn-benchmark根目录下面的datasets文...

    共 49 条
  • 1
  • 2
  • 3
  • 4
  • 5
  • 请选择