https://github.com/jahongir7174/efficientvitsam-onnx
EfficientViTSAM inference using ONNXRuntime
https://github.com/jahongir7174/efficientvitsam-onnx
efficientvitsam onnxruntime onnxruntime-gpu python sam
Last synced: 6 months ago
JSON representation
EfficientViTSAM inference using ONNXRuntime
- Host: GitHub
- URL: https://github.com/jahongir7174/efficientvitsam-onnx
- Owner: jahongir7174
- Created: 2024-02-16T09:47:39.000Z (over 1 year ago)
- Default Branch: master
- Last Pushed: 2024-02-21T01:40:48.000Z (over 1 year ago)
- Last Synced: 2025-04-15T19:04:51.950Z (6 months ago)
- Topics: efficientvitsam, onnxruntime, onnxruntime-gpu, python, sam
- Language: Python
- Homepage:
- Size: 870 KB
- Stars: 4
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
[EfficientViTSAM](https://github.com/mit-han-lab/efficientvit) inference using ONNXRuntime
### Installation
```
conda create -n SAM python=3.8
conda activate SAM
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
pip install onnxruntime
pip install PyYAML
pip install matplotlib
```
### Test
* Run `python main.py --image_path ./demo/cat.jpg --output_path ./demo/cat.png --mode point`
### Results
#### Reference
* https://github.com/mit-han-lab/efficientvit