チュートリアル通りで動くけど、一応メモ

環境

• windows 11
• python 3.10.5

動かす

学習済みモデルを使って、簡単なセマンティックセグメンテーションをしてみる。

もろもろインストールとダウンロード

pip install mmdet
pip install opencv-python
pip install torch torchvision torchaudio
pip install -U openmim
mim install mmcv-full
mim download mmdet --config yolov3_mobilenetv2_320_300e_coco --dest .

以下のようなpythonプログラムを書いて実行する

from mmdet.apis import init_detector, inference_detector

config_file = 'yolov3_mobilenetv2_320_300e_coco.py'
checkpoint_file = 'yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth'
model = init_detector(config_file, checkpoint_file, device='cpu')
img = 'cat.jpg'
result = inference_detector(model, img)
model.show_result(img, result, out_file='test_result.jpg')

cat.jpgは自分の画像を変更してください。