yolov7使用c++推理onnx模型cpu,gpu均可（附代码）-创新互联

  先放上原图以及推理效果图，这份数据集是我自己标注的，然后使用yolov7的官方代码进行训练得到了一个best.pt。最后还是用yolov7官方的export.py 导出得到一个onnx文件。文章下面我会附上数据集，需要自提。

yolov7环境搭建，模型的训练以及导出模型。

代码的编写已经文件的使用。

  我相信能能看到这里的朋友对于yolov7的使用应该是没有的问题的，那么只是需要使用官方代码导出一个onnx文件即可完成第一步的操作。

之后就会得到想要的onnx文件这里应该问题不大。

  当然cpu处理是非常简单有效的，只需要配置一个release版本的opencv即可。我这里重点讲述的是gpu版本的处理。

  1.首先要确定你的显卡型号，我的电脑是3060对应的cuda是11.1，cudnn8.1.1。当然我相信会使用yolov7的朋友对于这些是很熟悉的。我想强调的一点是由于vs平台的特殊性，vs平台的不同版本就对很多版本的cuda是不兼容的。例如vs2022对于老版本cuda就无法调用。同样老版本的vs对于新cuda也无法调用。我们在下载的时候一定要考虑一些版本适配性。

你是否还在寻找稳定的海外服务器提供商？创新互联www.cdcxhl.cn海外机房具备T级流量清洗系统配攻击溯源，准确流量调度确保服务器高可用性，企业级服务器适合批量采购，新人活动首月15元起，快前往官网查看详情吧