网创优客建站品牌官网
为成都网站建设公司企业提供高品质网站建设
热线:028-86922220
成都专业网站建设公司

定制建站费用3500元

符合中小企业对网站设计、功能常规化式的企业展示型网站建设

成都品牌网站建设

品牌网站建设费用6000元

本套餐主要针对企业品牌型网站、中高端设计、前端互动体验...

成都商城网站建设

商城网站建设费用8000元

商城网站建设因基本功能的需求不同费用上面也有很大的差别...

成都微信网站建设

手机微信网站建站3000元

手机微信网站开发、微信官网、微信商城网站...

建站知识

当前位置:首页 > 建站知识

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均可(附代码)-创新互联
网页URL:http://bjjierui.cn/article/epsdo.html

其他资讯