yolov3-tiny转tensorrt及bug解决
平台: Tesla P4,x86
环境信息:
ubuntu16.04
cuda9.0
cudnn7.5
python3.6
tensorflow-gou1.12
tensorrt5.1.5.0
onnx1.4.1(高版本会出现bug)
pycuda2019-1.12
流程: yolov3-tiny ----> tiny.onnx --->tiny.trt
darknet下yolov3-tiny的网络结构为:
layer filters size/strd
原创
2020-05-15 15:48:30 ·
1780 阅读 ·
0 评论