SPEED SAM C++ TENSORRT

🌐 1、概述
用于SAM
(segment anything model分割一切模型)的TensorRT
和CUDA
优化的高表现C++
实现,特别适用于实时图像分割任务。
📢 更新
- 模型转换:从ONNX模型构建TensorRT引擎以加速推理。
- 点和边界框分割:通过选定的点或边界框轻松分割图像。
- FP16精度:在速度与精度之间进行选择,支持FP16和FP32。
- 动态形状支持:利用优化配置文件高效处理可变输入大小。
- CUDA优化:利用CUDA进行预处理和有效的内存管理。
📢2、性能
2.1 推理时间
组件 |
SpeedSAM |
图像编码器 |
|
参数 |
5M |
速度< |