![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
DL
Nonentity_neu
不偏不倚,无过不及
展开
-
SegDecNet的多GPU数据同步训练 代码改动记录
论文:Segmentation-Based Deep-Learning Approach for Surface-Defect Detection环境:python3.5, cuda10 ,cudnn7.6 , tensorflow1.13参考:https://blog.csdn.net/jinxiaonian11/article/details/94316567记录原因:SegDecN...原创 2020-04-20 09:54:51 · 663 阅读 · 0 评论 -
TensorRT5.1.5.0入门 TensorRT头文件学习
目录结构NvInfer.hNvInferPlugin.h简介使用方法具体细节支持的层读取细节其他NvUtils.h简介使用方法具体细节NvCaffeParser.h最近时间宽裕,准备学习一下tensorRT的头文件。TensorRT-5.1.5.0includeNvInfer.hNvInferPlugin.hNvCaffeParser.hNvOnnxConfig.hNvOnn...原创 2019-08-17 10:43:33 · 4936 阅读 · 6 评论 -
Pytorch1.1.0 入门 自定义op(python)
nn.Parametertorch.nn.Parameterm,在看过很多博客的时候发现了一个用法self.v = torch.nn.Parameter(torch.FloatTensor(hidden_size)),首先可以把这个函数理解为类型转换函数,将一个不可训练的类型Tensor转换成可以训练的类型parameter并将这个parameter绑定到这个module里面(net.param...原创 2019-08-08 18:10:16 · 6255 阅读 · 1 评论 -
tensorRT5.1.5.0实践 EfficientNet+Pytorch的转换尝试即准确度测试
更新到2019.8.5参考EfficientNet-Pytorch.文章结构1. 翻译总结内容About EfficientNetAbout EfficientNet PyTorchUsage1. 翻译总结模块包含EfficientNet的op-for-op的pytorch实现,也实现了预训练模型和示例。这个代码是简单,而且高度可扩展的(extensible),并且容易集成到自己的项目...原创 2019-08-06 12:01:05 · 3614 阅读 · 7 评论 -
TensorRT5.1.5.0 实践 Pytorch2Onnx ,Onnx2TensorRT(python)
TensorRT5.1.5.x的关于onnx的demo只有两个,一个是c++的introductory_parser_samples,一个是python的introductory_parser_samples。参考了各方资料今天自己基于mnist写了个实例。环境: Ubuntu 16.06,python 3.5.2,torch 0.4.1, onnx 1.4.0 ,TensorRT 5.1.5...原创 2019-07-23 14:07:48 · 4770 阅读 · 4 评论 -
Onnx EfficientNet网络转onnx格式出现的问题记录
环境: ubuntu 16.04 python 3.5 torch 1.0.0 GTX1080拿到同事训练好的torch模型,想要保存为onnx格式,为以后转TensorRT做准备。代码就几行,torch.onnx定义的方法:import torchfrom torch.autograd import Variablefrom efficientnet import efficientn...原创 2019-07-26 15:49:13 · 4545 阅读 · 1 评论 -
TensorRT5.1.5.0入门 自定义层 IPlugin & IPluginV2的对比(C++)
最近研究 TensorRT的自定义曾,尝试的使用了自定义的FC层FC层和Upsample层Upsample层之后,重新回去看开发者手册,在此记录。 自己的理解,TensorRT的自定义层机制是有两个方法的,一种基于基类IPlugin,另一种是基于基类IPluginV2,从字面意思上来看IPluginV2就是最新版本。 IPlugin类的方法,是通过自己编写IPlugin的派生类IPlugi...原创 2019-07-30 13:31:37 · 4205 阅读 · 0 评论 -
TensorRT5.1.5.0实践 基于IPluginV2类的自定义层的用法
挖坑原创 2019-07-30 13:30:00 · 4378 阅读 · 2 评论 -
TensorRT5.1.5.0 实践 Doinference过程的探究(python)
挖坑原创 2019-07-20 12:25:54 · 6796 阅读 · 19 评论 -
TensorRT5.1入门 Object Detection With The ONNX TensorRT Backend In Python(python)
很快填坑原创 2019-07-18 14:03:53 · 1158 阅读 · 0 评论 -
TensorRT5.1入门 introductory_parser_samples
稍后更新原创 2019-07-18 11:12:50 · 626 阅读 · 0 评论 -
TensorRT5.1实践 自定义的Upsample层的方法
记录修改官方提供的trtexec,加入upsample层,跑检测网络的实践过程。接到一个类似yoloV3的检测网络,主要需要处理的是upsample自定义层。查询了一些资料,在github上下载了一些工程,终于是跑通了一个检测前馈时间的实例(还没有进行准确度测试)。...原创 2019-07-08 16:56:17 · 6158 阅读 · 4 评论 -
TensorRT5.1入门 sampleSSD
1.PPM格式图像知识来源:PPM文件格式详解(侵删,用作学习积累) PPM(Portable Pixmap Format)是源自PBM(位图bitmap,仅有黑与白,没有灰)和PGM(grayscale map,灰度图),PPM是通过RGB三种颜色显现的图像(pixmaps). 这三种图,无论哪种图像文件都是通过2个字节[magic number]来表明文件格式的类型以及编码方式(AS...原创 2019-06-24 16:25:50 · 2332 阅读 · 1 评论 -
Ubuntu16.04配置Caffe——Opencv3.1安装及Bug解决
环境:Ubuntu16.06 ,GTX 1080,CUDA 9.0,Cudnn 7.5.1 ,OPENCV 3.1.0(Deb)为了配置Caffe,为了以后着想,就一定要安装OpenCv。这是一个Caffe完整安装教程:传送门,大神写的非常清楚,在这里借鉴一下。这里只是想记录一下安装Caffe过程中,安装Opencv3.1.0期间出现的问题。(以下是传送门网址的原版叙述,侵删)一、下载...原创 2019-06-21 17:40:02 · 953 阅读 · 0 评论 -
TensorRT安装 & 环境配置
TensorRT安装&环境配置环境:ubuntu14.04, Tesla P4, Python3.4, Cuda8.0, Cudnn7.1.3 概述:从装有Ubuntu14.04和Python3.4的服务器上实现环境的配置。本文记录了在配置各个必要组件的环节中遇到的问题。1. 配置Nvidia驱动服务器显卡是Tesla的P4显卡,在2019.6.13这一天Nivida官网适用L...原创 2019-06-13 19:30:30 · 2492 阅读 · 0 评论