DeSeg是一个高性能的C++推理框架,它使用TensorRT无缝集成了YOLO目标检测和Segment Anything Model(SAM),适用于实时应用。 该框架支持交互式分割和由YOLO检测引导的自动分割,也可以在此框架上定义更多动作。
参考environment.md搭建环境。如果希望在windows系统下运行,请在win分支下查看环境配置教程。
main函数中的包含有测试代码的调用,如果希望有别的测试,请自行添加。 默认情况下,会调用sam模型的交互分割。
git clone https://github.com/curryc/DeSeg.git
cd DeSeg
mkdir build
cmake . build
cmake --build build
build/DeSeg
在弹出的窗口中,左键点击和拖拽即可进行交互分割。