shubham0204
/

sam2-onnx-models

Model card Files Files and versions

shubham0204 commited on Aug 18

Commit

83ff27a

•

1 Parent(s): 6bb8792

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -6,7 +6,9 @@ language:
 # SAM/SAM2 ONNX Models
-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.
 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.

 # SAM/SAM2 ONNX Models
+The ONNX models in this repository are converted using a **slightly modified** 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.
+Check the modified notebook here: https://github.com/shubham0204/Segment-Anything-Android/blob/main/notebooks/SAM2_ONNX_Export.ipynb
 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.