facebookresearch / facebookresearch/segment-anything

[enhance] export image encoder model to onnx

Open
#323 1 comment 3 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

In some scenarios, the image encoder may need to export onnx. I attempted to export it and deploy it using onnxruntime, which showed a significant improvement in speed. Can I submit a PR to add this feature?

Contributor guide

Open the contributing guide

Research direction

The issue names no files, tests, or entry points. First determine how the image encoder is exposed for inference and what export path is expected; done means a working ONNX export and ONNX Runtime deployment path, with the reported speed improvement verified.

Written by the indexing model from the issue text.

Assessment

Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.