【文件属性】:
文件名称:tensorRT-Caffe:在NVIDIA Jetson TX2上使用tensorRT的演示加速了AlexNet的Caffe模型
文件大小:2.1MB
文件格式:ZIP
更新时间:2021-05-02 19:13:23
caffe nvidia tensorrt tx2 C
张量RT-Caffe
在NVIDIA Jetson TX2上使用tensorRT的演示加速了AlexNet的Caffe模型。
有关中文的详细说明,请参考NVIDIA JETSON TX2 tensorRT加速Caffe实战.pdf 。
先决条件:
NVIDIA Jetson TX2
CUDA 8.0
人工神经网络
张量RT
.prototxt文件
.caffemodel文件
.binaryproto文件
建议您使用Jetpack 3.1刷新TX2设备,以便可以自动安装所有必需的工具。
我们使用的Caffe模型
我们尝试对三种不同类型的停车位进行分类:
因此,我们在Caffe中使用了Alexnet来实现此任务。
输入和输出由Caffe模型的prototxt文件指定:
layer {
name: "data"
type: "Input"
top: "data"
【文件预览】:
tensorRT-Caffe-master
----stb_image.h(254KB)
----type.txt(2B)
----imgs()
--------ParkingSlotType1.bmp(88KB)
--------resize.jpeg(10KB)
--------ParkingSlotType2.bmp(88KB)
--------ParkingSlotType3.bmp(88KB)
--------caffetrt.png(86KB)
--------CAFFE.png(101KB)
----giexec.cpp(7KB)
----NVIDIA JETSON TX2 tensorRT 加速 Caffe 实战.pdf(1.79MB)
----Makefile.giexec(4KB)
----README.md(4KB)
----Makefile(116B)
----file.txt(40B)