facebookresearch / facebookresearch/segment-anything
ONNX example is not a really full ONNX example
- Dominant language
- Jupyter Notebook
- Stars
- 54.9k
- Forks
- 6.4k
- PR merge metrics
- No merged PRs in 30d
Description
As I notice in example we use exported onnx model only to do prediction, but generation of embeddings still done using pytorch and python.
Which is very limited approach if for example I want to deliver to the system that uses onnx runtime only.
Is there a way to export onnx for calculating embeddings as well?
Contributor guide
Research direction
Start by reviewing the repository's ONNX example and the current PyTorch/Python embedding-generation path. Determine how embeddings are produced and what export boundary the example uses. Done means the example supports generating embeddings through an ONNX Runtime-only workflow, rather than using PyTorch and Python for that step.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook, python, pytorch
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100