Skip to content

lyxlplhy/Sam2Onnx_Inference

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Sam2Onnx_Inference

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

参考

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published