https://github.com/lyxlplhy/sam2onnx_inference
https://github.com/lyxlplhy/sam2onnx_inference
cpp onnx sam2
Last synced: 10 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/lyxlplhy/sam2onnx_inference
- Owner: lyxlplhy
- Created: 2024-11-22T09:08:52.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2025-02-19T06:39:39.000Z (over 1 year ago)
- Last Synced: 2025-04-03T20:03:12.310Z (over 1 year ago)
- Topics: cpp, onnx, sam2
- Language: C++
- Homepage:
- Size: 10.2 MB
- Stars: 13
- Watchers: 1
- Forks: 1
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Sam2Onnx_Inference
SAM2分割实现文件包含Model.h、Sam2.h、Sam2.cpp和测试代码main.cpp，支持float32、float16推理
## 功能：
* Sam2单帧照片，单目标分割，使用点或者框作为提示信息(节省推理效率，去除了视频推理中记忆模块的使用)
* 输入多个提示点、框同时分割单张照片中多目标代码调试中，位于 [Multi-object-segmentation](https://github.com/lyxlplhy/Sam2Onnx_Inference/edit/Multi-object-segmentation/)
* yolo目标检测+sam2分割
* Sam2所使用onnx权重格式生成代码[Sam2Onnx_Inference](https://github.com/lyxlplhy/Sam2-collection?tab=readme-ov-file#onnx%E5%AF%BC%E5%87%BA)
## 依赖
* opencv=4.8
* onnxruntime=1.18.0
* cuda=12.4
* cudnn
## 参考
* Sam2:[https://github.com/facebookresearch/sam2](https://github.com/facebookresearch/sam2)
* OrtInference：[https://github.com/Aimol-l/OrtInference?tab=readme-ov-file](https://github.com/Aimol-l/OrtInference?tab=readme-ov-file)
* ONNX-SAM2-Segment-Anything:[https://github.com/ibaiGorordo/ONNX-SAM2-Segment-Anything](https://github.com/ibaiGorordo/ONNX-SAM2-Segment-Anything)
* Sam2-collection: [https://github.com/lyxlplhy/Sam2-collection/tree/master](https://github.com/lyxlplhy/Sam2-collection/tree/master)