SAM C++ TensorRT（实时图像分割）

于 2025-02-19 15:26:17 发布
阅读量584 收藏 5
点赞数 4
CC 4.0 BY-SA版权
分类专栏： 计算机视觉 文章标签： TensorRT CUDA C++ 图像分割
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/old_power/article/details/145730444

SPEED SAM C++ TENSORRT

在这里插入图片描述

🌐 1、概述

用于SAM（segment anything model分割一切模型）的TensorRT和CUDA优化的高表现C++实现，特别适用于实时图像分割任务。

📢 更新

  • 模型转换：从ONNX模型构建TensorRT引擎以加速推理。
  • 点和边界框分割：通过选定的点或边界框轻松分割图像。
  • FP16精度：在速度与精度之间进行选择，支持FP16和FP32。
  • 动态形状支持：利用优化配置文件高效处理可变输入大小。
  • CUDA优化：利用CUDA进行预处理和有效的内存管理。

📢2、性能

2.1 推理时间

组件 SpeedSAM
图像编码器
参数 5M
速度<
最低0.47元/天 解锁文章
SAM-TensorRT模型转换
whaosoft143ai的博客
06-20 612
Segment Anything Model (SAM)模型包含三个组件，如图1图像编码器提示编码器和掩码解码器。图1：分割一切模型（SAM）概述。重量级图像编码器输出图像嵌入，然后可以通过各种输入提示有效地查询，以平摊的实时速度产生对象掩码。对于对应于多个对象的模糊提示，SAM可以输出多个有效的掩码和相关的置信度分数。Image encoder图像编码器。受到可扩展性和强大的预训练方法的启发，使用了一个MAE预训练的Vision Transformer（ViT）最小化地适应处理高分辨率输入whaosoft
使用 C++ 中的 TensorRT 实现 SAM
最新发布
12-13
使用 C++ 中的 TensorRT 实现 SAM 名为SPEED-SAM-C++-TENSORRT 的项目是 Segment Anything Model (SAM) 的高性能实现，使用 NVIDIA 的 TensorRT 进行高效推理，并使用 CUDA 优化 GPU 利用率。
分割一切模型 Fast SAM C++推理部署---TensorRT （有核心代码）
勿忘初心,方得始终
08-12 2648
使用C++部署分割一切模型FastSAM
使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip
05-15
使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip
硬核详解Segment Anything Model (SAM) TensorRT模型转换
lovely_yoshino的博客
06-12 1万+
是Facebook Research近来开源的一种新的图像分割任务、模型，该模型被设计和训练为可提示的，因此它可以将zero-shot transfer零样本迁移到新的图像分布和任务。其分割效果较为惊艳，是目前分割SOTA的算法。关于该算法的详细细节网上有很多的解释，本文主要分享如何将该模型转换为TensorRT的模型，方便后期部署加速模型推理。Segment Anything Model (SAM)模型包含三个组件，如图1图像编码器提示编码器和掩码解码器。图1：分割一切模型（SAM）概述。
0778-极智开发-解读YOLOv6及示例代码
03-05
0778_极智开发_解读YOLOv6及示例代码
基于TensorRT的C++高性能推理库源码+使用说明(支持RT-DETR，Yolov5v7v8等).zip
01-09
1.项目代码均经过功能验证ok，确保稳定可靠运行。欢迎下载体验！下载完使用问题请私信沟通。 2.主要针对各个计算机相关专业，包括计算机科学、信息安全、数据科学与大数据技术、人工智能、通信、物联网等领域的在校学生、专业教师、企业员工。 3.项目具有丰富的拓展空间，不仅可作为入门进阶，也可直接作为毕设、课程设计、大作业、初期项目立项演示等用途。 4.当然也鼓励大家基于此进行二次开发。在使用过程中，如有问题或建议，请及时沟通。 5.期待你能在项目中找到乐趣和灵感，也欢迎你的分享和反馈！ 【资源说明】 基于TensorRT的C++高性能推理库源码+使用说明(支持RT-DETR，Yolov5、v7、v8、yoloX、OSTrack、LightTrack等) Highlights - 支持全景驾驶感知 YOLOPv2，目标检测 RT-DETR，Yolo 5/X/7/8 ，多目标跟踪 Bytetrack，单目标跟踪 OSTrack、LightTrack； - 预处理和后处理实现CUDA核函数，在 jetson 边缘端也能高性能推理； - 封装Tensor、Infer，实现内存复用、CPU/GPU 内存自动拷贝、引擎上下文管理、输入输出绑定等； - 推理过程实现生产者消费者模型，实现预处理和推理的并行化，进一步提升性能； - 采用 RAII 思想+接口模式封装应用，使用安全、便捷。 Easy Using 本项目代码结构如下：`apps` 文件夹中存放着各个算法的实现代码，其中 `app_xxx.cpp` 是对应 `xxx` 算法的调用demo函数，每个算法彼此之间没有依赖，假如只需要使用yolopv2，可以将此文件夹下的其他算法全部删除，没有影响；`trt_common` 文件夹中包括了常用的cuda_tools，对TensorRT进行Tensor、Infer的封装，生产者消费者模型的封装；`quant-tools` 文件夹中是量化脚本，主要是yolov5/7；`workspace` 文件夹中存放编译好的可执行文件、engine等。 使用哪个算法就在 `main.cpp` 中调用哪个算法的demo函数。 ```bash . ├── apps │   ├── yolo │   └── yolop │   ├── app_yolo.cpp │   ├── app_yolop.cpp │   ├── ... ├── trt_common │   ├── cuda_tools.hpp │   ├── trt_infer.hpp │   ├── trt_tensor.hpp │   └── ... ├── quant-tools │   └── ... ├── workspace │ └── ... ├── CMakeLists.txt └── main.cpp ``` 如果要进行您自己的算法部署，只需要在 `apps` 文件夹中新建您的算法文件夹，模仿其他算法中对 `trt_infer/trt_tensor` 等的使用即可。后续时间空闲较多的情况下会更新较为详细的用法。 更多详见项目说明
使用tensorRT推理加速
A1983Z的博客
11-08 387
tensorrt推理加速
SAM2图像分割项目+跑通版本+segment-anything-2.zip
08-02
【内容概要】: 本资源包含SAM2（Segment Anything Model 2）图像分割项目的完整跑通版本，压缩包命名为`segment-anything-2.zip`。该项目利用先进的深度学习技术实现高效、精确的图像实例分割。压缩包内含预训练模型...
【课程设计】使用TensorRT部署SAM分割一切大模型C++源码+部署步骤.zip
05-16
【课程设计】“使用TensorRT部署SAM分割一切大模型C++源码+部署步骤”是一个针对深度学习模型优化与部署的实践项目，旨在帮助学生或开发者掌握如何利用TensorRT高效地运行复杂的大规模模型，特别是涉及到语义分割...
算法部署-使用TensorRT部署RT-DETR目标检测算法-优质算法部署项目实战.zip
10-20
算法部署_使用TensorRT部署RT-DETR目标检测算法_优质算法部署项目实战
SAM后缀自动机
10-12
SAM后缀自动机，陈立杰2012年noi冬令营的讲稿，带你了解后缀自动机
学习记录——语义分割、实时分割和全景分割的区别、几个Norm的区别、Dice 、BCE
Chaoy6565的博客
07-18 936
语义分割（Semantic Segmentation）：语义分割是将图像分割成多个语义区域的任务，即将图像中的每个像素分配给特定的语义类别。每个像素都被标记为属于不同的类别，例如人、汽车、树等。语义分割关注的是像素级别的分类，目标是理解图像中不同物体的位置和边界。实时分割（Real-Time Segmentation）：实时分割是指能够在实时或接近实时的速度下进行图像分割的任务。实时分割通常要求在处理连续的视频帧时保持高帧率，以实现实时的交互或应用。
【计算机视觉】中科院发布Fast SAM，精度相当SAM，速度提升了50倍！
wzk4869的博客
06-26 3614
【计算机视觉】中科院发布Fast SAM，精度相当SAM，速度提升了50倍！
后缀自动机(SAM)——黑盒使用方案
R的博客
08-05 530
重在SAM的性质和使用方法而不对过于繁杂的原理深究。
使用5行代码的实时图像分割
深度学习与计算机视觉
01-16 3524
目录计算机视觉应用中的图像分割图像分割批处理图像分割视频分割摄像机视频的分割计算机视觉应用中的图像分割计算机视觉是计算机看到和分析他们所看到的东西的能力。图像分割是计算机视觉的一个方面，它...
Segment-Anything TensorRT部署------onnx转engine踩坑
weixin_38423729的博客
12-05 407
SAM模型TensorRT部署及pytorch转onnx转engine文件踩坑
mean shift 图像分割（一、二、三）
热门推荐
u011511601的专栏
06-02 2万+
MeanShift图像分割算法：大概是将复杂的背景，通过粗化提取整体信息，进而将图像分割。 接下来我想，将会抽出一部分时间，研究一下这个算法，以最终实现手势形状提取。 《Mean Shift: A Robust Approach Toward Feature Space Aalysis》一文中，利用Meanshift算法分割图像，大体类似于这样的效果： 看到一篇非常好哒博文
【深度学习】YOLOv5实例分割 数据集制作、模型训练以及TensorRT部署
qq_40198848的博客
11-19 1万+
yolov5 seg
如何微调SAM模型进行专业图像分割任务
标签中强调了“图像分割”，这可能意味着文件内容与如何利用SAM模型进行图像分割有关，也可能包含了如何通过代码实现模型的微调。图像分割是理解、解释和处理图像内容的基础，因此它在多个领域如医疗成像分析、自动...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值