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这种跨连接的层是不是可以直接实现的。

猜你喜欢

转载自www.cnblogs.com/yanxingang/p/10851116.html
今日推荐