tensorRT 与yolov3_tiny

Posted yanxingang

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorRT 与yolov3_tiny相关的知识,希望对你有一定的参考价值。

yolov3-tiny中有下面这些层:

Convolutional

Maxpooling

Leaky-Relu

Linear-Relu(正常的Relu)

Residual Block

Strided Residual Block

Upsample

查看TensorRT支持的网络层种类:

https://developer.download.nvidia.cn/compute/machine-learning/tensorrt/docs/5.1/rc/TensorRT-Support-Matrix-Guide.pdf

ONNX的格式是都支持上面的层的,只是不清楚Residual Block这种跨连接的层是不是可以直接实现的。

 

以上是关于tensorRT 与yolov3_tiny的主要内容,如果未能解决你的问题,请参考以下文章

转换后的 Tensorrt 模型与 Tensorflow 模型的输出形状不同?

TensorRT 与 TensorFlow 1.7 集成

TensorRT安装与engine生成

TensorRT安装与engine生成

TensorRT安装与engine生成

tensorRT 与yolov3_tiny