EfficientSAM-ONNX-Sample

yformer/EfficientSAM をColaboraotry上でONNX推論するサンプルです。

Demo.mp4

Usage

ONNX変換、推論テスト

Colaboratoryでノートブックを開き、上から順に実行してください。

簡易デモ

以下コマンドでデモを起動してください。
左クリックでプロンプト座標を追加、右クリックでプロンプト座標を削除します。
また、キーボード1～3で、クリック時に追加する座標のタイプを変更します（1：対象座標、2：非対象座標、3：対象バウンディングボックス）

python demo.py

• --image
  画像ファイルの指定
  デフォルト：sample.png
• --encoder
  エンコーダーONNXファイルのパス
  デフォルト：onnx_model/efficient_sam_vitt_encoder.onnx
• --decoder
  デコーダーONNXファイルのパス
  デフォルト：onnx_model/efficient_sam_vitt_decoder.onnx

Note

サンプルの画像はぱくたそ様の「トゲトゲのサボテンとハリネズミ」を使用しています。

License

EfficientSAM-ONNX-Sample is under Apache-2.0 license.

Author

高橋かずひと(https://twitter.com/KzhtTkhs)