模型实战(17)之C++ - tensorRT部署yolov8seg实例分割

2023-12-21 23:48:36

模型实战(17)之C++ - tensorRT部署yolov8seg实例分割

  • yolov8是2023年以来最新的集检测、分割、分类、关键点于一身的模型
  • 参考官网文档:添加链接描述
  • 本文将详细给出C++ - tensorRT部署yolov8seg模型的详细步骤
  • 实现效果的话,精度和python环境下差不多,仅推理时间在RTX30~系列显卡仅需几毫秒一帧,但是其前处理和后处理时间在图像尺寸为640时达到了一百多毫秒,有点慢了
  • 后期也将进行优化:通过cuda实现模型输入输出的前处理与后处理!!!
  • 环境:Win10 cuda11.7 tensorrt8.4.3 cudnn860 RTX3050

在这里插入图片描述

1. Python环境下调用

  • 首先,下载好yolov8模型的源码:https://github.com/ultralytics/ultralytics
  • 然后搭建好yolov8的运行环境
  • 最后执行下边预测命令:
yolo predict model=

文章来源:https://blog.csdn.net/yohnyang/article/details/135141382
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。