tensorRT 与yolov3_tiny
yolov3-tiny中有下面这些层:
Convolutional
SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。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这种跨连接的层是不是可以直接实现的。
如果采用的是C++ 的API,有下面几个问题思考:
1、weight格式如何转换成为ONNX的格式
2、如何读取图片,输入网络
3、网络的输出,转换为准确度
如果采用的是Python的API,那么直接就会有Yolo-V3的示例。

更多精彩