Windowsで OpenMMLab の MMagic を使って Matting

環境

WIndows 11
CUDA 11.7
Python 3.10

Python環境構築

pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 --index-url https://download.pytorch.org/whl/cu117
pip install mmcv==2.0.1 -f https://download.openmmlab.com/mmcv/dist/cu117/torch2.0.0/index.html
pip install openmim==0.3.9
pip install mmagic==1.0.2
pip install albumentations==1.3.1

Trimap画像の作成

適当なペイントソフトで作ることになります。

左が元画像、右がTrimap画像です。

今回は公式で用意してくれているサンプルです。

グレースケールで外側：0、境界：128、内側：255で動作確認できました。

実行

python demo/mmagic_inference_demo.py ^
  --model-name gca  ^
  --img ./resources/input/matting/GT05.jpg ^
  --trimap ./resources/input/matting/GT05_trimap.jpg ^
  --result-out-dir ./resources/demo_matting_gca_res.png

結果

かなり境界を正確に検出してくれます。

一般のSegmentationモデルより正確にSegmentationできると思います。