Skip to content

Latest commit

 

History

History
4 lines (4 loc) · 383 Bytes

README.md

File metadata and controls

4 lines (4 loc) · 383 Bytes

修改测试代码,不用做项目,不支持更新,只用作学习,参考自各位大佬的代码,

2.1 配置环境变量:安装TensorRT,cuda,cudnn,pycuda,UFF,graphsurgeon,onnx(解析器)

2.2 转换yolo的weights和cfg文件为trt文件

2.3 添加脚本并修改onnx_to_tensorrt.py文件,输出yolo和tensorrt对比时间,输出识别结果图片