2020.03.17Facebook人工知能が開発している最新の物体検出アルゴリズムを実装しているソフトウェアシステムDetectron2のインストールメモ。Detectron2をインストールする前にPyTorchをインストールする。Networksが自前のフレームワークChainerの開発をやめて、PyTorchに乗り換えたことで日本でも注目を集めている。Detectron2のインストールはpipなのでとても簡単。R-CNN)をかけたもの。Truck以外はほぼ正解。torch.cuda.is_available()True>>>Detectron2本記事はメモなので以下の説明に従ってインストール等をすることを勧める。pycocotoolsインストールインストール済みのCUDAのバージョンが10.2だったので以下のコマンドでソースからビルドした。10.1に対応している。detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl成功すると下図のようなウインドウが表示される。demo.pyの使い方は次のコマンドで調べることができる。detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl訓練済みモデルによる推論デモ(プログラムによる実行)上のコマンドの代わりにプログラムで実行するためには以下のようなコードが必要となる。cv2.destroyAllWindows()このコードをtest.pyとして~/src/detectron2/myprogディレクトリ下に保存する。myprogフォルダはないので事前に次のコマンド作成する。~/src/detectron2/myprog次のコマンドを実行すると上図のウインドウが表示される。ウインドウ上にマウスのポインタをもっていき、何かキーを押すとプログラムは終了する。コメントをキャンセルメールアドレスが公開されることはありません。人気記事ODE初級講座ホーム検索トップサイドバータイトルとURLをコピーしました。