ONNX
English
shubham0204 commited on
Commit
bd3141b
·
verified ·
1 Parent(s): 50af02d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +38 -3
README.md CHANGED
@@ -1,3 +1,38 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ ---
6
+
7
+ # SAM/SAM2 ONNX Models
8
+
9
+ The ONNX models in this repository are converted using the Colab notebook present in [github.com/ibaiGorordo/ONNX-SAM2-Segment-Anything](https://github.com/ibaiGorordo/ONNX-SAM2-Segment-Anything). Use the `sam2/sam2.py` or `annotation_app.py` scripts in the `ONNX-SAM2-Segment-Anything` repository to perform inference with the ONNX models.
10
+
11
+ The converted ONNX models are stored in this repository to facilitate their usage in the [github.com/shubham0204/Segment-Anything-Android](https://github.com/shubham0204/Segment-Anything-Android) project, which uses the `onnxruntime` in Android to infer the SAM/SAM2 models.
12
+
13
+
14
+ ## Citations
15
+
16
+ ```text
17
+ @misc{ravi2024sam2segmentimages,
18
+ title={SAM 2: Segment Anything in Images and Videos},
19
+ 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},
20
+ year={2024},
21
+ eprint={2408.00714},
22
+ archivePrefix={arXiv},
23
+ primaryClass={cs.CV},
24
+ url={https://arxiv.org/abs/2408.00714},
25
+ }
26
+ ```
27
+
28
+ ```text
29
+ @misc{kirillov2023segment,
30
+ title={Segment Anything},
31
+ 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},
32
+ year={2023},
33
+ eprint={2304.02643},
34
+ archivePrefix={arXiv},
35
+ primaryClass={cs.CV},
36
+ url={https://arxiv.org/abs/2304.02643},
37
+ }
38
+ ```