随笔分类 -  pytorch

摘要:网络定义 import torch as torch import torch.nn as nn class LeNet(nn.Module): def __init__(self): super(LeNet,self).__init__() layer1 = nn.Sequential() lay 阅读全文
posted @ 2020-08-20 15:29 无左无右 阅读(1126) 评论(0) 推荐(0) 编辑
摘要:通过索引赋值 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.Te 阅读全文
posted @ 2020-06-29 10:12 无左无右 阅读(757) 评论(0) 推荐(0) 编辑
摘要:参考: Building Autoencoders in Keras[https://blog.keras.io/building-autoencoders-in-keras.html] Keras上实现AutoEncoder自编码器[https://blog.csdn.net/marsjhao/a 阅读全文
posted @ 2020-06-28 09:20 无左无右 阅读(379) 评论(0) 推荐(0) 编辑
摘要:pytorch/libtorch qq群: 1041467052 模型部署到现场为了防止泄密,需要加密。加密一方面可以防止泄密,另一方面可以便于模型跟踪管理,防止混淆。 libtorch的加载模型的函数,torchjitload();我点开load可以看到函数。有两个: TORCH_API std: 阅读全文
posted @ 2020-06-24 13:10 无左无右 阅读(3414) 评论(0) 推荐(0) 编辑
摘要:pytorch/libtorch qq群: 1041467052 本文介绍的是直接读取caffe权重,并根据caffe网络一步一步的搭同样的pytorch网络。本文以分割网络segnet为例。 首先给出参考链接: https://blog.csdn.net/DumpDoctorWang/articl 阅读全文
posted @ 2020-06-11 21:39 无左无右 阅读(571) 评论(0) 推荐(0) 编辑
摘要:pytorch/libtorch qq群: 1041467052 首先,需要掌握libtorch的一些语法,可以参考下面的链接: [https://www.cnblogs.com/yanghailin/p/12901586.html] 大概说下pytorch转libtorch流程: 1.先训练pyt 阅读全文
posted @ 2020-05-26 15:02 无左无右 阅读(902) 评论(0) 推荐(0) 编辑
摘要:pytorch/libtorch qq群2群:302984355 pytorch/libtorch qq群: 1041467052(一群满了) 其实pytorch的函数libtorch都有,只是写法上有些出入。 libtorch的官方文档链接 class tensor 只是官方文档只是类似与函数申明 阅读全文
posted @ 2020-05-16 18:14 无左无右 阅读(39337) 评论(1) 推荐(6) 编辑
摘要:1.error: conversion from ‘atTensorAccessor<float, 2ul, atDefaultPtrTraits, long int>’ to non-scalar type ‘at::Tensor’ requested torch::Tensor result_d 阅读全文
posted @ 2020-05-12 16:13 无左无右 阅读(4589) 评论(1) 推荐(0) 编辑
摘要:pytorch/libtorch qq群: 1041467052 有一点儿感悟就是: 一定要去官网找一手资料,百度出来的都是个人根据官网来的,这个就随便看看了。 官网:https://github.com/pytorch/pytorch 首先我跑的是pytorch 1.3版本的,conda安装好的, 阅读全文
posted @ 2020-04-29 11:06 无左无右 阅读(5463) 评论(2) 推荐(0) 编辑
摘要:深度学习ocr交流qq群:1020395892 论文地址:https://link.zhihu.com/?target=https%3A//arxiv.org/pdf/1911.08947.pdf github:https://github.com/MhLiao/DB 捣鼓DB有一段时间了,年前就开 阅读全文
posted @ 2020-03-03 14:24 无左无右 阅读(5098) 评论(3) 推荐(0) 编辑
摘要:深度学习ocr交流qq群:1020395892 论文地址:https://link.zhihu.com/?target=https%3A//arxiv.org/pdf/1911.08947.pdf github:https://github.com/MhLiao/DB 捣鼓DB有一段时间了,年前就开 阅读全文
posted @ 2020-02-20 21:23 无左无右 阅读(1205) 评论(0) 推荐(0) 编辑
摘要:本文转摘于如下链接: 逆卷积的详细解释ConvTranspose2d(fractionally strided convolutions) https://www.cnblogs.com/wanghui garcia/p/10791328.html pytorch官方手册:https://pytor 阅读全文
posted @ 2020-02-19 15:48 无左无右 阅读(2733) 评论(0) 推荐(0) 编辑
摘要:Deformable ConvNets v1: 论文地址:https://github.com/chengdazhi/Deformable Convolution V2 PyTorch 工程地址:https://github.com/felixlaumon/deform conv 论文地址: Def 阅读全文
posted @ 2020-02-17 15:06 无左无右 阅读(12673) 评论(0) 推荐(2) 编辑
摘要:官方:https://github.com/MhLiao/DB 周军大神实现的:https://github.com/WenmuZhou/DBNet.pytorch 1.官方的 官方的按照安装流程很容易安装,只是我的环境是ubuntu16.04+cuda8,所以一直用的pytorch1.0.1(py 阅读全文
posted @ 2020-01-18 17:31 无左无右 阅读(1020) 评论(5) 推荐(0) 编辑
摘要:[[1.] [0.] [1.] [0.] [1.] [1.] [0.] [0.]] 0代表相似,1代表不相似 loss曲线: 测试: 数据集: https://i.cnblogs.com/Files.aspx 阅读全文
posted @ 2019-09-30 11:28 无左无右 阅读(2251) 评论(1) 推荐(0) 编辑
摘要:1.模型保存与加载 1.1 #a、保存 推荐仅仅保存模型的state_dict torch.save(model.state_dict(), MODELPATH) # .pt .pth #b、加载 model = TheModelClass(*args, **kwargs) model.load_s 阅读全文
posted @ 2019-09-29 12:00 无左无右 阅读(248) 评论(0) 推荐(0) 编辑
摘要:错误1: 2019 07 16 10:02:35,438 maskrcnn_benchmark.trainer INFO: Start training /opt/conda/conda bld/pytorch_1549633347309/work/aten/src/THCUNN/ClassNLLC 阅读全文
posted @ 2019-07-16 14:07 无左无右 阅读(1741) 评论(0) 推荐(0) 编辑
摘要:评价模型指标,自带的eval_ctw1500.py是根据两两框重叠面积大于0.5算正例tp。其中用到了import Polygon as plg 模块,可以方便的处理多边形的重叠计算面积 1.pip install Polygon2 2.cover = set() 这里用到了set,之前在c++里面 阅读全文
posted @ 2019-07-05 16:25 无左无右 阅读(979) 评论(0) 推荐(0) 编辑
摘要:1 阅读全文
posted @ 2019-06-14 15:03 无左无右 阅读(1094) 评论(0) 推荐(0) 编辑