Skip to content

Latest commit

 

History

History
20 lines (17 loc) · 967 Bytes

File metadata and controls

20 lines (17 loc) · 967 Bytes

DeSeg:以tensorRT为基础的实时检测分割框架

DeSeg是一个高性能的C++推理框架,它使用TensorRT无缝集成了YOLO目标检测和Segment Anything Model(SAM),适用于实时应用。 该框架支持交互式分割和由YOLO检测引导的自动分割,也可以在此框架上定义更多动作。

快速开始

参考environment.md搭建环境。如果希望在windows系统下运行,请在win分支下查看环境配置教程。

build & run

main函数中的包含有测试代码的调用,如果希望有别的测试,请自行添加。 默认情况下,会调用sam模型的交互分割。

git clone https://github.com/curryc/DeSeg.git
cd DeSeg
mkdir build
cmake . build
cmake --build build
build/DeSeg

在弹出的窗口中,左键点击和拖拽即可进行交互分割。