TensorRT上首次运行demo（sampleSSD）笔记

其实官方已经有了非常详细的教程，https://github.com/NVIDIA/TensorRT；https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/sampleSSD。但是由于我对神经网络很是不熟悉，所以导致理解有误。

设置CUDA_INSTALL_DIR 和 CUDNN_INSTALL_DIR（我设置的是/usr/lib/aarch64-linux-gnu/，官网提供的镜像里并没有和cuda类似的目录，只有lib，没有头文件，不太清楚）。
只是想运行一下Demo的话，可以在设置好CUDA_INSTALL_DIR 和CUDNN_INSTALL_DIR之后直接看Prerequisites。
需要下载的资源 models_VGGNet_VOC0712_SSD_300x300。上述网站上提供的链接国内不能直接下载，我在网上没有找到免费的，于是找别人帮忙下载的。应该也会保留一段时间，有需要的可以私信我。
Edit the deploy.prototxt file and change all the Flatten layers to Reshape operations with the following parameters:
所有type是Flatten ，修改为Reshape，参数也按如下修改。

layer {name: "conv6_2_mbox_conf_flat"type: "Reshape"bottom: "conv6_2_mbox_conf_perm"top: "conv6_2_mbox_conf_flat"reshape_param {shape {dim: 0dim: -1dim: 1dim: 1}
}
}

Update the detection_out layer to add the keep_count output as expected by the TensorRT DetectionOutput Plugin. top: “keep_count”
意思是新增加一句。我开始理解错了。报错：Plugin layer output count is not equal to caffe output count.
Some tactics do not have sufficient workspace memory to run.
运行出现这一句，不要担心，我也出现了，好像并不影响。等待稍长一会，就会有结果。
运行成功截图。

TensorRT上首次运行demo（sampleSSD）笔记相关推荐

【深度学习】动漫风格迁移AnimeGANv2，发布线上运行Demo
内容一览:最强二次元风格迁移模型 AnimeGAN 更新啦,现在可以在线上轻松运行模型,还可以调整风格参数,输出你想要的照片效果. 关键词:风格迁移机器视觉二次元 AnimeGANv2 最近发布了 ...
动漫风格迁移 AnimeGANv2 ，发布线上运行 Demo
作者 | 神经星星出品 | HyperAI超神经 By 超神经内容一览:最强二次元风格迁移模型 AnimeGAN 更新啦,现在可以在线上轻松运行模型,还可以调整风格参数,输出你想要的照片效果. 关 ...
动漫风格迁移AnimeGANv2，发布线上运行Demo
来源:HyperAI超神经本文约800字,建议阅读5分钟最强二次元风格迁移模型 AnimeGAN 更新啦,现在可以在线上轻松运行模型,还可以调整风格参数,输出你想要的照片效果. 关键词:风格迁移 ...
DuiLib学习笔记1.编译运行demo
c++中皮肤问题比较麻烦,MFC自带的太难用.DirectUI界面库就比较强大了,之前像skin++之类的基于DirectUI收费昂贵.DuiLib是基于DirectUI的界面库,可以将用户界面和处理 ...
历史上的今天：XML 之父诞生；第一台存储程序计算机首次运行；IBM 推出 AS/400...
整理 | 王启隆透过「历史上的今天」,从过去看未来,从现在亦可以改变未来. 今天是 2022 年 6 月 21 日,在 22 年前的今天,中国联通在香港.纽约成功上市,筹资 56.5 亿美元,进入全 ...
【历史上的今天】6 月 21 日：第一台存储程序计算机首次运行；XML 之父诞生；IBM 推出 AS/400
整理 | 王启隆透过「历史上的今天」,从过去看未来,从现在亦可以改变未来. 今天是 2022 年 6 月 21 日,在 22 年前的今天,中国联通在香港.纽约成功上市,筹资 56.5 亿美元,进入全 ...
《MYSQL是怎样运行的》笔记|配置文件|系统变量|字符集|InnoDB存储结构|数据页结构|索引结构与使用|数据目录|表空间|连表原理|查询优化|BufferPool|事务|redo与undo|锁
<MYSQL是怎样运行的>笔记前记: 历时15天,笔记+看书.完成于2022.2.5. 本书是讲具体的数据库实现,而数据库系统概念见:https://blog.csdn.net/qq_4 ...
cuda8+cuDNN Faster R-CNN安装塈运行demo
安装cuda cuda8安装参见网上教程安装cuDNN py-faster-rcnn/caffe-fast-rcnn目前不支持cuDNN5. 如果使用cuDNN5,在编译caffe-fast-rcn ...
大数据开发工程师基本功修炼之史上最全Linux学习笔记(建议)
我正在参加年度博客之星评选,请大家帮我投票打分,您的每一分都是对我的支持与鼓励. 2021年「博客之星」参赛博主:Maynor大数据 (感谢礼品.红包免费送!) https://bbs.csdn.ne ...

TensorRT上首次运行demo（sampleSSD）笔记

TensorRT上首次运行demo（sampleSSD）笔记相关推荐

最新文章

热门文章