首页 > 编程语言 >yolov8分割法 C++部署

yolov8分割法 C++部署

时间:2024-06-12 12:13:38浏览次数:13  
标签:分割 -- onnx tensorrt C++ yolov8 seg 640 报错

 

使用的命令:

conda list

 

参考资料 https://github.com/triple-Mu/YOLOv8-TensorRT/blob/main/docs/Segment.md

1.

python3 export-seg.py --weight ./0.0.0/yolov8s-seg.pt --opset 11 --sim --input-shape 1 3 640 640 --device cuda:0

报错:ModuleNotFoundError: No module named 'tensorrt'

解决:

pip install nvidia-tensorrt

 

2.

python3 export-seg.py --weight ./0.0.0/yolov8s-seg.pt --opset 11 --sim --input-shape 1 3 640 640 --device cuda:0

报错:The NVIDIA driver on your system is too old

解决:

将520版本的驱动更换为535的驱动

 

3.

python3 build.py --weights yolov8s-seg.onnx --fp16  --device cuda:0 --seg
报错:
AttributeError: 'tensorrt_bindings.tensorrt.IBuilderConfig' object has no attribute 'max_workspace_size'
原因:tensorrt8.0以上删除了max_workspace_size属性。

解决方案:(1)将tensorRT降低为7.x版本(不推荐)

(2)使用如下命令替换:
# builder.max_workspace_size = 1 << 20
config = builder.create_builder_config()
config.max_workspace_size = 1 << 20


4.trtexec --onnx=yolov8n-seg.onnx --saveEngine=yolov8n-seg.trt
报错:
[8] Assertion failed: inputs.at(0).isInt32() && "For range operator with dynamic inputs, this version of TensorRT only supports INT32!"
解决:
3060比较特殊,不要用trtexec 编译onnx,应该使用代码编译
https://github.com/FeiYull/TensorRT-Alpha/blob/main/tools/onnx2trt.cpp

标签:分割,--,onnx,tensorrt,C++,yolov8,seg,640,报错
From: https://www.cnblogs.com/liuweiweitju/p/18243679

相关文章

  • 图集分割软件(绿色、纯净、免费、共享)
    推荐一款自己编写的一款软件,图集分割软件(绿色、纯净、免费、共享)因为开发的时候需要将一些图集软件还原成碎图。也找到可以用的,但都不够理想:texturepack软件成生的图集没有相应的碎图工具,倒是有一些可用的,spine生成的图集本生自带碎图工具,操作麻烦,所以想自己动手写一个可以同......
  • 计算机毕业设计项目推荐,32127 爬虫-自驾游搜索系统(开题答辩+程序定制+全套文案 )上万套
    目 录摘要1绪论1.1研究背景1.2爬虫技术1.3flask框架介绍21.4论文结构与章节安排32 自驾游搜索系统分析42.1可行性分析42.2系统流程分析42.2.1数据增加流程52.3.2数据修改流程52.3.3数据删除流程52.3系统功能分析52.3.1功能性分析62.......
  • 计算机毕业设计项目推荐,32006 node 中国传统节日介绍网站(开题答辩+程序定制+全套文案
    基于node.js中国传统节日介绍网站 摘 要随着科学技术的飞速发展,社会的方方面面、各行各业都在努力与现代的先进技术接轨,通过科技手段来提高自身的优势,中国传统节日介绍网站当然也不能排除在外。中国传统节日介绍网站是以实际运用为开发背景,运用软件工程原理和开发方法,采......
  • 计算机毕业设计项目推荐,29042 基于Web的医院护理管理系统的设计(开题答辩+程序定制+全
    摘 要随着科学技术的飞速发展,社会的方方面面、各行各业都在努力与现代的先进技术接轨,通过科技手段来提高自身的优势,医院当然也不例外。医院预约管理系统是以实际运用为开发背景,运用软件工程原理和开发方法,采用Java技术构建的一个管理系统。整个开发过程首先对软件系统进......
  • C/C++ 宏定义注意事项
    在C/C++中,宏定义是通过#define预处理指令来实现的。宏定义虽然方便,但也有一些潜在的问题和注意事项需要开发者注意。以下是关于C/C++宏定义相关的注意事项:没有类型检查:宏定义是文本替换,所以编译器不会进行类型检查。这可能导致在替换后产生类型不匹配或意外的行为。括......
  • C/C++学学习计划
    一、自我认知最近高考的结束让考生学子们顺利走完了一段艰苦的学习生涯,意味着可以暂时放下学习去享受生活,看着他们散发青春活力的样子我突然想到了当时的我也是那个模样。但回过头来,我已经是一个大二快结束,下半年就大三的学生了,我对未来充满着焦虑,对未来的方向也十分迷茫,我自......
  • C / C++ 保留两位小数(setprecision(n)的一些用法总结)
    转载:https://blog.csdn.net/qq_36667170/article/details/79265224做题遇到保留两位小数的题目,课本上写的又多又杂,网上查来的也是一堆内容需要筛选,눈_눈还是自己总结一下吧。首先说C++代码 #include<iomanip>//不要忘了头文件 //第一种写法 cout<<setiosflags(io......
  • C++基础入门学习记录
    本系列基于黑马程序员|c++课程,记录学习相关视频——黑马程序员匠心之作|C++教程从0到1入门编程,学习编程不再难_哔哩哔哩_bilibiliC++基础入门2数据类型C++规定在创建一个变量或者常量时,必须要指定出相应的数据类型,否则无法给变量分配内存2.1整型作用:整型变量表示的是=......
  • CH01_初识C++
    CH01_初识C++第一个C++程序新建项目新建文件编写代码#include<iostream>usingnamespacestd;intmain(){ cout<<"helloworld"<<endl; system("pause"); return0;}运行程序注释作用:在代码中加一些说明和解释,方便阅读代码。两种格式:单行注释://这是......
  • RecyclerView如何设置分割线颜色
    importandroid.content.Context;importandroid.content.res.TypedArray;importandroid.graphics.Canvas;importandroid.graphics.Paint;importandroid.graphics.Rect;importandroid.graphics.drawable.Drawable;importandroid.view.View;importandroidx.core.content.......