非常感谢https://github.com/wang-xinyu/tensorrtx作者无私贡献。
环境
- Ubuntu18.04
- cuda10.2、cudnn8.x.x
- tensorrt-7.x.x.x
- opencv >= 3.3
- 支持 Jetson TX2
yolov4:
https://github.com/wang-xinyu/tensorrtx/tree/master/yolov4
yolov5
https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5