Introduction
nvidia dockerを使えるようになったので、UbuntuでもYOLOv3を試してみます。
Get Started
今回は、自分でDockerHubにイメージをpushしてみました。
dockerfileもgithubに登録してあります。
dockerfileの中身はシンプルに
- apt-getで必要なバイナリ収集
- OpenCVをビルドしてインストール
- darknetのビルド
を行っています。
下記のコマンドで、イメージをpullし、コンテナに入れます。
実行後は、コンテナは破棄されます。
1 | sudo nvidia-docker run --rm -i -t takuyatakeuchi/yolo-darknet:9.1-cudnn7-devel-ubuntu16.04 /bin/bash |
コンテナに入ったら、weightファイルを拾ってきて、実行します。
1 | cd /tmp/darknet |
Windows版と遜色ない性能です。
Docker経由でも性能は変わりません。
CUIでの実行なので、本来はOpenCV経由で画像が画面に表示されますが、それはエラーになっています。
ただし、出力結果はディスクにpredictions.jpgとして出力されていますので確認できます。
scpコマンドで画像ファイルを転送するなりして確認します。
1 | scp ./predictions.jpg XXXX@:~ |