bukuroo commited on
Commit
2b049c8
·
verified ·
1 Parent(s): 97a4837

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -3
README.md CHANGED
@@ -1,3 +1,33 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - onnx
5
+ - segmentation
6
+ - sam2
7
+ - ezonnx
8
+ ---
9
+ ### EdgeTAM
10
+ ONNX models for inference with [EZONNX](https://github.com/ikeboo/ezonnx)
11
+
12
+
13
+ - Model type:
14
+ Segmentation
15
+
16
+ - Official repository:
17
+ [EdgeTAM](https://github.com/facebookresearch/EdgeTAM)
18
+ - Setup
19
+ ```
20
+ pip install git+https://github.com/ikeboo/ezonnx.git
21
+ ```
22
+ - Usage
23
+ ```python
24
+ from ezonnx import EdgeTAM, visualize_images
25
+ tam = EdgeTAM()
26
+ res = tam.set_image("images/cat.jpg")
27
+ # add point with positive or not and its label
28
+ res = tam.set_point((240, 180), True, 1)
29
+ res = tam.set_point((50,50), True, 2)
30
+ res = tam.set_point((400,50), True, 2)
31
+ res = tam.set_point((150, 330), True, 3)
32
+ visualize_images("Masked Image",res.visualized_img)
33
+ ```