Python
Chainer
chainercv

ChainerCV YOLO 動かしてみた

要約

  • ChainerCVでもYOLOを利用できるようになった
  • 2018/05/16時点ではソースコードからインストールを行う必要がある
  • YOLOv2・YOLOv3の2種類が準備されている

概要

  • ChainerCV(github)にてYOLOが利用できるようになったらしいので動かしてみた

環境

  • OS: Windows 10 1803 WSL Ubuntu18.04
  • python: 3.6.5
  • pip: 10.0.1

準備

  • 2018/05/16時点の最新バージョンである0.9.0にはまだ組み込まれていなかった
  • リポジトリではYOLOのモデルを見つけられたのでソースコードからのインストールを試みた
$ git clone https://github.com/chainer/chainercv
$ cd chainercv
$ pip install -e .

認識

  • どこかで見たことのある以下の画像を認識させた

sample.jpg

YOLOv2

$ python examples/yolo/demo.py sample.jpg

sample_yolov2.jpeg

YOLOv3

$ python examples/yolo/demo.py sample.jpg --model yolo_v3

sample_yolov3.jpeg