/EfficientViTSAM-onnx

EfficientViTSAM inference using ONNXRuntime

Primary LanguagePython

EfficientViTSAM 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