SAM(https://segment-anything.com/)を使ってマスク画像を作成するツールです。
- https://github.com/facebookresearch/segment-anything からViT-H SAM modelをダウンロードしてフォルダ直下に保存
- segment-anythingの実行環境を構築 (pip install git+https://github.com/facebookresearch/segment-anything.git)
- gui_auto.pyファイルのimpathsに入力画像へのパスを設定してください。
- gui_auto.pyファイルのdeviceに使用するデバイスをしてしてください。（GPUがなければ"cpu"に変更）
下記のコマンドで実行できます。
python gui_auto.py画面が3つ表示されます。
- 元画像
- SAMセグメンテーション画像
- マスク画像
すべての画面に共通の下記のコマンドを使って、マスク画像を作成します。
- 左クリック：クリックした領域をマスクに追加
- 右クリック：クリックした領域をマスクから消す
- sキー：マスク画像をmaskフォルダに保存
- rキー：マスク画像を初期化
- nキー：次の画像へ
- 1キー：領域選択モード
- 2キー：塗り絵モード