Is there any plan to add support for SAM2.1

[Anivie]

Any plan to add support for SAM2.1?

[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

[GzuPark]

@ryouchinsa Thank you. It works well.