https://github.com/jahongir7174/efficientvitsam
EfficientViTSAM inference using PyTorch
https://github.com/jahongir7174/efficientvitsam
efficientvitsam python pytorch sam segment-anything-model
Last synced: 3 months ago
JSON representation
EfficientViTSAM inference using PyTorch
- Host: GitHub
- URL: https://github.com/jahongir7174/efficientvitsam
- Owner: jahongir7174
- Created: 2024-02-15T05:20:42.000Z (over 1 year ago)
- Default Branch: master
- Last Pushed: 2024-02-21T01:31:37.000Z (over 1 year ago)
- Last Synced: 2025-04-10T14:16:38.794Z (7 months ago)
- Topics: efficientvitsam, python, pytorch, sam, segment-anything-model
- Language: Python
- Homepage:
- Size: 878 KB
- Stars: 8
- Watchers: 0
- Forks: 0
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
[EfficientViTSAM](https://github.com/mit-han-lab/efficientvit) inference using PyTorch
### 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 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