Is there any code for model export？

[XYZ-916]

Such as onnx or torchscript

[ryouchinsa]

You can export SAM 2.1 ONNX models for image prediction using export_onnx.py in sam-cpp-macos repository.

Install Segment Anything Model 2 and download checkpoints.

Copy yaml files in sam2/configs/sam2.1 to sam2.

Put export_onnx.py and david-tomaseti-Vw2HZQ1FGjU-unsplash.jpg to the root folder.

To export an ONNX model.

python export_onnx.py --mode export

To check how the ONNX model works.

python export_onnx.py --mode import

You can download exported SAM 2.1 ONNX models.

• SAM 2.1 Tiny
• SAM 2.1 Small
• SAM 2.1 BasePlus
• SAM 2.1 Large