SAM2分割实现文件包含Model.h、Sam2.h、Sam2.cpp和测试代码main.cpp,支持float32、float16推理
- Sam2单帧照片,单目标分割,使用点或者框作为提示信息(节省推理效率,去除了视频推理中记忆模块的使用)
- 输入多个提示点、框同时分割单张照片中多目标代码调试中,位于 Multi-object-segmentation
- yolo目标检测+sam2分割
- Sam2所使用onnx权重格式生成代码Sam2Onnx_Inference
- opencv=4.8
- onnxruntime=1.18.0
- cuda=12.4
- cudnn
- Sam2:https://github.com/facebookresearch/sam2
- OrtInference:https://github.com/Aimol-l/OrtInference?tab=readme-ov-file
- ONNX-SAM2-Segment-Anything:https://github.com/ibaiGorordo/ONNX-SAM2-Segment-Anything
- Sam2-collection: https://github.com/lyxlplhy/Sam2-collection/tree/master