网站建设资讯

NEWS

网站建设资讯

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

成都创新互联-企业级云服务器提供商,为用户提供云服务器、CDN、云安全服务、重庆服务器托管、服务器租用、高防服务器租用等全方位云服务与各行业解决方案,帮助企业及个人极速备案,轻松上云,安全无忧。

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

要点:

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

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

一.模型导出

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

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

二.vs平台环境搭建 

  当然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元起,快前往官网查看详情吧


当前名称:yolov7使用c++推理onnx模型cpu,gpu均可(附代码)-创新互联
转载来源:http://cdweb.net/article/epsdo.html