【Deep Learning】Keras-yolo3で画像認識

とりあえずKeras-yolo3を動かして画像認識を行うところまでを説明します。

準備として以下ようにgitをクローンします。
カレントディレクトリに新しくフォルダが出来ます。

git clone https://github.com/qqwweee/keras-yolo3.git

新しく出来たフォルダに移動してください。

cd keras-yollo3

wgetで本家サイトからモデルファイルを取得します。

wget https://pjreddie.com/media/files/yolov3.weights

kerasのフォーマットに変換します。既にスクリプトは用意されています。

python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5

以下コマンドで実行。
問題なければ画像のパスを入力します。

python yolo_video.py --image

実際に画像認識をした結果がこちらです。

認識結果

精度やアノテーション位置としては申し分ないです。
次回は詳しいプログラムの内容等を説明しようかと思います。