yoraba build

備忘録を兼ねた技術ブログ

SSD画像解析

 短い秋が過ぎ去り、身も心も凍えさせる乾いた風が吹き始めた頃。ふと、動画解析がしたくなりしました。

 まずデータを調達するところから始まるわけですが、色々と品を変えたい場合はネットから拾うのが便利という発想からyoutubeを選択しました。

 今回使った動画は懐かしCMです。CMって色々な人や物が目まぐるしく登場するので、画像解析の題材として優れていると思います。  

 ソースはコピペの嵐です。

 OpenCVで解析して緑色の枠を作っている部分はお世辞にも精度が良いとは言えません。

 キャプション付きの赤いバウンティングボックスを書いている部分はPyTorchのSSDサンプルソースで解析したものです。車と人を認識してくれています。画像が荒いせいか、所々識別されていませんが、見切れていても判別してくれるのは凄いです。上では表示していませんが、他には、すりおろされているリンゴとかを認識してくれました。

 当然、セガサターンは認識してくれませんでした。