在篇二中通过数十个步骤在没有CMakeLists.txt的前提下,通过自己创建的VS工程成功实现了yolov5的C++推理代码的编译工作。接下来就是生成tensorRT的.engine权重和实现推理。
需要注意的是:1. python版本的tensorRT中模型是需要预热的,C++可以不用。2. tensorRT输出的engine权重文件只适用于当前算力的硬件设备,不同算力是不支持的,因此需要在部署的机器上编译生成该文件才行。
- 步骤一:打开篇二中自己定义的VS项目的输出目录,输入cmd并回车。