SAM/SAM2 ONNX 模型

本仓库中的 ONNX 模型是使用 github.com/ibaiGorordo/ONNX-SAM2-Segment-Anything 中提供的 经过轻微修改的 Colab notebook 转换而成的。请使用 ONNX-SAM2-Segment-Anything 仓库中的 sam2/sam2.py 或 annotation_app.py 脚本对这些 ONNX 模型进行推理。

在此查看修改后的 notebook：https://github.com/shubham0204/Segment-Anything-Android/blob/main/notebooks/SAM2_ONNX_Export.ipynb

转换后的 ONNX 模型存储在本仓库中，以便在 github.com/shubham0204/Segment-Anything-Android 项目中使用，该项目使用 Android 上的 onnxruntime 对 SAM/SAM2 模型进行推理。

引用

@misc{ravi2024sam2segmentimages,
      title={SAM 2: Segment Anything in Images and Videos}, 
      author={Nikhila Ravi and Valentin Gabeur and Yuan-Ting Hu and Ronghang Hu and Chaitanya Ryali and Tengyu Ma and Haitham Khedr and Roman Rädle and Chloe Rolland and Laura Gustafson and Eric Mintun and Junting Pan and Kalyan Vasudev Alwala and Nicolas Carion and Chao-Yuan Wu and Ross Girshick and Piotr Dollár and Christoph Feichtenhofer},
      year={2024},
      eprint={2408.00714},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2408.00714}, 
}

@misc{kirillov2023segment,
      title={Segment Anything}, 
      author={Alexander Kirillov and Eric Mintun and Nikhila Ravi and Hanzi Mao and Chloe Rolland and Laura Gustafson and Tete Xiao and Spencer Whitehead and Alexander C. Berg and Wan-Yen Lo and Piotr Dollár and Ross Girshick},
      year={2023},
      eprint={2304.02643},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2304.02643}, 
}