LoginSignup
18
17

More than 5 years have passed since last update.

物体検出で遊んでみた♬

Last updated at Posted at 2018-04-10

ここまで画像やってきたので、いよいよ物体検知♬

有名どころのもので、遊んでみた
SSD: Single Shot MultiBox Detector 高速リアルタイム物体検出デモをKerasで試す

Kerasバージョンを探せ

そして、オリジナルは以下のサイト
weiliu89/caffe
でも、やはりKerasの方が見やすい
rykov8/ssd_keras
ところが、これKerasのバージョンが
This code was tested with Keras v1.2.2, Tensorflow v1.0.0, OpenCV v3.1.0-dev
ということで、バグを探っていくと、。。。
たどり着けました(笑)

以下が今回実行できたバージョンのサイト
cory8249/ssd_keras
This code was tested with Keras v1.2.2, Tensorflow v1.0.0, OpenCV v3.1.0-dev
Also support newest Keras v2.0.1 (using ssd_v2.py)

少しだけカスタマイズ

画像を少し追加して、物体検知後の画像を保存できるように改変して。。。
肝心な変更として、ACCがもとは0.6以上としてますが、案外低いのも当たっているので、0.3に変更しました。
これでめでたく、人間が識別したのと似たような結果画像が得れました。

それだけですが、以下のとおり物体検知できました‼

コードは以下に置きました。
MuAuan/SSD

なお、フィッティング済重みは以下からダウンロードできます。
Weights are ported from the original models and are available here. You need weights_SSD300.hdf5,

結果

recognized_picture001.png
recognized_picture002.png
recognized_picture003.png
recognized_picture004.png
recognized_picture005.png
recognized_picture006.png
recognized_picture007.png
recognized_picture009.png

18
17
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
18
17