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 .

認識

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

• 認識は公式のサンプルコード(わずか46行!!)により行った

YOLOv2

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

YOLOv3

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