facebookresearch / facebookresearch/segment-anything

ONNX example is not a really full ONNX example

Open
#16 13 comments 22 reactions 0 assignees View on GitHub
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.