8 无左无右

尚未进行身份认证

我要认证

好记性不如破键盘---点滴、积累、进步!

等级
TA的排名 1w+

转libtorch验证精度小程序

pytorch端和libtorch端转的每一步都需要验证精度,一般情况下,就debug看看两天的tensor值最开始的或者结尾一不一样,一般相差10的-4就没有问题,实在不行就根据形状随机的写下标,写几个看看值是否差不多。下面具体说明:法1:就是debugpycharm界面可以看到tensor的值,但是也只能看到前几个,libtorch端debug看不到具体的值,就需要打印,比如下图...

2020-09-14 20:10:00

cmakelist 添加第三方库

在libtorch工程中添加一个dcn的库,编译出来的库名字是/data_4/everyday/0902/snake/snake-refinenet/dcn_cpp_plugin/myinstall/lib/libdcn_v2_cuda_forward_v2.so下面是完成的总的完成的正确的:cmake_minimum_required(VERSION 3.0 FATAL_ERRO...

2020-09-02 16:23:00

pytorch model

目录网络定义model.named_children 返回名字 和 操作model.modules() 可用于参数初始化其他的可以参考:model.parameters() || torch.optim.SGD(params, lr=, momentum=0, dampening=0, weight_decay=0, nesterov=False)[source]网络定义impo...

2020-08-20 15:29:00

python pytorch opencv代码积累

目录print两种写法打印pytorch权重python os.path globnp.unique(b) numpy 去重根据标签图和颜色表显示标签图print两种写法print('[Epoch: %d, numImages: %5d]' % (epoch, i * self.args.batch_size + image.data.shape[0])) print("Acc:...

2020-08-15 10:42:00

交叉熵 pytorch中的nn.CrossEntropyLoss()函数

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

2020-08-13 17:31:00

voc分割数据集 pil调色板

voc分割数据集有两种,文件夹名字分别是SegmentationClass,SegmentationClassAug,其中SegmentationClass文件夹图片样式如下:SegmentationClassAug文件夹图片样式如下:今天来说下SegmentationClass文件夹带彩色图的,读一个deeplab的pytorch代码的时候,我就在找是怎么把颜色对应到标签图的,...

2020-08-13 13:52:00

deeplab系列论文研读

Deeplab v1:(2015)SEMANTIC IMAGE SEGMENTATION WITH DEEP CONVOLUTIONAL NETS AND FULLY CONNECTED CRFSDeeplab v2:(2016.06)Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution...

2020-07-28 20:18:00

感受野计算

本文参考在卷积神经网络中,感受野的定义是:卷积神经网络每一层输出的特征图(feature map)上的像素点在原始图像上映射的区域大小。"""striden = strides= stride1*stride2*...*striden-1rn = rn-1 + (fn - 1)*strides# 卷积核大小/stride/padding# 卷积核大小/stride/pa...

2020-07-21 09:24:00

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....

2020-06-29 10:12:00

tensorflow1.15 keras 自编码 -- > Keras上实现AutoEncoder自编码器

参考:Building Autoencoders in Keras[https://blog.keras.io/building-autoencoders-in-keras.html]Keras上实现AutoEncoder自编码器[https://blog.csdn.net/marsjhao/article/details/68928486]全连接import tensorflo...

2020-06-28 09:20:00

libtorch 模型加密

模型部署到现场为了防止泄密,需要加密。加密一方面可以防止泄密,另一方面可以便于模型跟踪管理,防止混淆。libtorch的加载模型的函数,torch::jit::load();我点开load可以看到函数。有两个:TORCH_API std::shared_ptr<script::Module> load(const std::string& filename, ...

2020-06-24 13:10:00

caffe模型转pytorch

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

2020-06-11 21:39:00

pytorch refinedet libtorch实现

首先,需要掌握libtorch的一些语法,可以参考下面的链接:[https://www.cnblogs.com/yanghailin/p/12901586.html]大概说下pytorch转libtorch流程:1.先训练pytorch的模型,并测试2.把pytorch模型转pt3.写后处理2.把pytorch模型转pt这个可以单独写个脚本,也可以在跑测试脚本的时候在其中某个位置加上两句话就可以了。单独写脚本例子如下:import torchfrom net import resnet

2020-05-27 09:07:02

keras 训练保存网络图,查看loss,acc,单张图片推理,保存中间feature map图,查看参数...

目录训练脚本,同时打印网络结构,保存了网络图和loss,acc图,保存训练的模型加载模型(这里只加载模型文件包括了网络),单张图片预测显示中间某层的feature map比如看conv2d_1 (Conv2D) (None, 28, 28, 32) 这个的feature map同样的,我需要看pool层之后的feature map同样的,我需要看conv2d_2层之...

2020-05-23 11:39:00

libtorch 常用api函数示例(史上最全、最详细)

其实pytorch的函数libtorch都有,只是写法上有些出入。libtorch的官方文档链接:https文章目录调试技巧:CMakeLists.txt0.torch::full_like1.创建与初始化tensor 1.1 torch::rand 1.2 torch::empty 1.3 torch::ones 1.4 torch::Tensor keep = torch::zeros({scores.size(0)}).to(torch::kLong).to(scores.device())

2020-05-18 18:54:35

libtorch 报错与解决

1.error: conversion from ‘at::TensorAccessor<float, 2ul, at::DefaultPtrTraits, long int>’ to non-scalar type ‘at::Tensor’ requestedtorch::Tensor result_data = result_.accessor<float,2&g...

2020-05-12 16:13:00

ubuntu 源码编译libtorch

有一点儿感悟就是:一定要去官网找一手资料,百度出来的都是个人根据官网来的,这个就随便看看了。官网:https://github.com/pytorch/pytorch首先我跑的是pytorch 1.3版本的,conda安装好的,现在需要源码编译。按照官网的流程,需要先安装依赖包。我是切换到pytorch1.3的conda虚拟环境进行的:这里是官网的教程:其中,git clon...

2020-04-29 11:06:00

遗传算法 商旅问题 c++ GA tsp

#include <stdio.h>#include <string.h>#include <stdlib.h>#include "math.h"#include "time.h"#define CITY_NUM 38 //城市数,城市编号是0~CITY_NUM-1#define POPSIZE 300 //种群个体...

2020-04-26 19:49:00

神经网络的复杂度分析

本博客主要转载于如下链接:https://zhuanlan.zhihu.com/p/31575074 卷积神经网络的复杂度分析http://www.360doc.com/content/17/0719/14/10408243_672570496.shtml vgg16参数量与计算量首先分析vgg:网络图vgg16详细参数将所有的memory加起来:大约就是24M,卷积的...

2020-04-17 16:07:00

源码编译opencv的安卓Android库

因为项目需要,稍微改了opencv一点源码,这个时候就需要源码编译opencv的库了。我pc机器上面试了没有问题,但是我们项目是跑在移动设备安卓上面的,算法通过jni打包接口供上层java调用,opencv用的安卓版本的,这就坑爹了,没办法,我就编译安卓版本的opencv吧,这里面的坑真多。参考链接:https://www.jianshu.com/p/925614d6e9e0 ...

2020-04-13 13:13:00

查看更多

勋章 我的勋章
  • 阅读者勋章Lv1
    阅读者勋章Lv1
    授予在CSDN APP累计阅读博文达到3天的你,是你的坚持与努力,使你超越了昨天的自己。
  • 持之以恒
    持之以恒
    授予每个自然月内发布4篇或4篇以上原创或翻译IT博文的用户。不积跬步无以至千里,不积小流无以成江海,程序人生的精彩需要坚持不懈地积累!
  • 勤写标兵Lv1
    勤写标兵Lv1
    授予每个自然周发布1篇到3篇原创IT博文的用户。本勋章将于次周周三上午根据用户上周的博文发布情况由系统自动颁发。
  • 分享精英
    分享精英
    成功上传11个资源即可获取