Fast R-CNN:ディープラーニングによる一般物体検出手法

R-CNNに続き、Fast R-CNNのアルゴリズムについて勉強しよう。


ディープラーニングによる一般物体検出アルゴリズムまとめ
これまで勉強したディープラーニングによる一般物体検出手法のアルゴリズム一覧。


再びこちらの系譜図を引用する↓

系譜図にはR-CNNFast R-CNNの間にSPPnetというのがあるのでSPPnetについて軽く触れておく。


スポンサーリンク

SPPnet

SPPnetは、ECCV2014で発表された論文 Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognitionで提案された手法。

R-CNNではSelective Searchで検出した領域候補をそれぞれ個別にCNNにかけて特徴抽出していたのに対し、SPPnetでは画像全体をCNNに入力して全体の特徴マップを抽出する。互いに重なる部分も多い2000個の矩形領域を個別にCNNにかける冗長性を排除したわけですね。

SPPnetの処理の全体像はこちらの図が分かりやすい↓

SPPnet

SPP(Spatial Pyramid Pooling)

元画像からSelective Searchで検出した矩形に対応する特徴マップの領域にSpatial Pyramid Pooling(空間ピラミッドプーリング)と呼ばれるpooling処理を施す。このSpatial Pyramid Poolingにより、矩形領域のサイズに関わらず固定長のベクトルを得ることができる。

Spatial Pyramid Poolingでは、特徴マップを異なるサイズのウィンドウ(16×16, 4×4, 1×1)でmax-poolingした結果を平坦化して連結し、固定長のベクトルにしてから次の全結合層の入力とする。

そして、全結合層からの出力をR-CNNと同様にSVMで分類(Classification)、回帰(Regression)でBounding Boxを求める。

SPPnetの欠点

R-CNNに比べてCNNによる特徴抽出の処理は効率化したが、特徴量を抽出する畳み込み層、全結合層、候補領域を分類するSVMやBounding Boxの回帰などは別々に学習しなければならないまま。

Fast R-CNN

R-CNNと同じ著者(Ross Girshick氏)がICCV2015で発表したFast R-CNNは、画像全体からCNN(ここではVGGを利用)で特徴マップを抽出するまではSPPnetと同様。違うのはこちらのスライドで言うと3~5のプロセス↓


スポンサーリンク

ここからはRegion Proposal(候補領域)という言葉に代わってRegion of Interest(関心領域)という言葉が使われるけど、分類(Classification)対象となる領域を指しているので同じ意味です。

RoI (Region of Interest) Pooling

Fast R-CNNでは、SPP(Spatial Pyramid Pooling)のピラミッドレベルを1つに簡略化したRoI(Region of Interest)Poolingによって検出領域を固定サイズのベクトルに収める。

RoI Pooling

RoI Poolingは、アスペクト比の違いを考慮してMax Poolingを行う処理。
こちらのgifが分かりやすいか(オープンソースのRoIプーリング実装のページの画像)↓

RoI pooling in TensorFlow

ニューラルネットワークの分岐

RoI Poolingによって固定サイズ化された領域は、全結合層(Fully connected layers)を経てRoI feature vector(RoI特徴ベクトル)となる。このRoI feature vectorを、以下2つの全結合層の入力とする。

  • 候補領域(物体)の分類(Classification)タスクを学習する全結合層
  • Bounding Boxの回帰(Regression)タスクを学習する全結合層

つまり、ここからニューラルネットワークが2つに分岐し、タスクごとの全結合層を経て出力層へ向かう。

RoI feature vector

Multi-task Loss

物体の分類とBounding Boxの回帰という2つのタスク(Multi-task)を学習するために、この2のタスクの推定誤差を同時に考慮した損失関数(loss function) Multi-task Lossが提案された。

Multi-task Lossを最小化するようニューラルネットワークを学習することで、2つのタスクを同時に習得させることができる。

ここで改めてFast R-CNNのニューラルネットワークの構造を見てみると、R-CNNSPPnetでは特徴抽出のCNNと分類や回帰の学習が独立していたのに対し、Fast R-CNNでは分類・回帰の学習も同じニューラルネットワークとして繋がっていることが分かる↓

Fast R-CNN

特徴抽出からタスクの学習までが1つのニューラルネットワークに繋がったことで、紫の矢印のように誤差逆伝搬で学習時にネットワークの全てのパラメーターが更新されるようになった。

Fast R-CNNの欠点

さて、特徴抽出から分類・回帰までを一貫したニューラルネットワークにしたことで、R-CNNの頃よりも遥かに効率化しました。
しかし、冒頭の系譜図にもあったようにFast R-CNNNOT End-to-Endなアプローチに分類される理由は、RoIを求める処理がニューラルネットワークとは独立してSelective Searchで行われているからなのです。

次はFaster R-CNNについてまとめようと思う。


ディープラーニングによる一般物体検出アルゴリズムまとめ
これまで勉強したディープラーニングによる一般物体検出手法のアルゴリズム一覧。


スポンサーリンク

関連記事

UnrealCV:コンピュータビジョン研究のためのUnreal Engineプラグイン
.NETで使えるTensorFlowライクなニューラルネットワークライブラリ『NeuralNetwo...
MeshroomでPhotogrammetry
CGのためのディープラーニング
PCA (主成分分析)
SSII 2014 デモンストレーションセッションのダイジェスト動画
Mask R-CNN:ディープラーニングによる一般物体検出・Instance Segmentatio...
CGAN (Conditional GAN):条件付き敵対的生成ネットワーク
SONY製のニューラルネットワークライブラリ『NNabla』
ディープラーニングに対応したPythonの機械学習ライブラリ『Pylearn2』
pythonの機械学習ライブラリ『scikit-learn』
DCGAN (Deep Convolutional GAN):畳み込みニューラルネットワークによる敵...
Two Minute Papers:先端研究を短時間で紹介するYouTubeチャンネル
PGGAN:段階的に解像度を上げて学習を進めるGAN
Super Resolution:OpenCVの超解像処理モジュール
PyDataTokyo主催のDeep Learning勉強会
Open3D:3Dデータ処理ライブラリ
Structure from Motion (多視点画像からの3次元形状復元)
OpenCVの三角測量関数『cv::triangulatepoints』
COLMAP:オープンソースのSfM・MVSツール
Faster R-CNN:ディープラーニングによる一般物体検出手法
Adobeの手振れ補正機能『ワープスタビライザー』の秘密
Multi-View Stereoによる3次元復元ライブラリ『OpenMVS』
U-Net:ディープラーニングによるSemantic Segmentation手法
iPhoneで3D写真が撮れるアプリ『seene』
疑似3D写真が撮れるiPhoneアプリ『Seene』がアップデートでついにフル3Dモデルが撮影できる...
OpenCV 3.1とopencv_contribモジュールをVisual Studio 2015で...
CycleGAN:ドメイン関係を学習した画像変換
C#で使える遺伝的アルゴリズムライブラリ『GeneticSharp』
手を動かしながら学ぶデータマイニング
SSII2014 チュートリアル講演会の資料
Active Appearance Models(AAM)
OpenCV 3.1のsfmモジュールを試す
Unityで強化学習できる『Unity ML-Agents』
Windows10でPyTorchをインストールしてVSCodeで使う
openMVG:複数視点画像から3次元形状を復元するライブラリ
NumSharp:C#で使えるNumPyライクな数値計算ライブラリ
Google製オープンソース機械学習ライブラリ『TensorFlow』のWindows版が公開された
今年もSSII
オープンソースのStructure from Motionツール『Regard3D』
Kinect for Windows V2のプレオーダー開始
機械学習で流体シミュレーションを近似する『Physics Forests』

コメント