Detectronの使い方

まずm208にログインする。

次にdockerを起動する。

sudo nvidia-docker run --name yoshitake01 -it detectron:c2-cuda9-cudnn7 /bin/bash

yoshitake01のところは適当な名前に変更

Detectronによる画像認識

cd /detectron
python2 tools/infer_simple.py --cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml --output-dir /tmp/detectron-visualizations --image-ext jpg --wts https://s3-us-west-2.amazonaws.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl demo

/detectron/demoの中にjpgを入れるとそれらを認識した結果を/tmp/detectron-visualizationsに出力してくれる。

例えば、のような結果が得られる。 1枚の画像から複数の物体を手軽に検出できて良さそう。

ドローンの画像でも可能だった。