web-dev-qa-db-ja.com

TensorFlowの高速RCNN

TensorFlowバージョンのFRCNNを実装している人はいますか?次のような関連リポジトリが見つかりました。

  1. ROIプールレイヤーの実装
  2. py-faster-rcnnリポジトリに基づく高速RCNNの実装

しかし、1のために:ROIプーリング層が動作すると仮定し(私は試していません)、次のように実装する必要があるものがあります:

  • ROIデータレイヤー(例: roidb
  • 線形回帰 SmoothL1Loss
  • エンドツーエンドトレーニングのROIプールレイヤーの後処理。ROIプールレイヤーの結果を分類器のCNNにフィードするように変換する必要があります。

2:em ....では、Caffeに基づいて前処理(たとえばroidb)を準備し、データをTensorflowにフィードしてモデルをトレーニングするpy-faster-rcnnに基づいているようです。それ。

それで、私が知りたいのは、 Tensorflowが将来的にはより高速なRCNNをサポートする ?です。そうでない場合、上記の誤解はありますか?またはそれをサポートするレポや誰かがいますか?

44
RyanLiu

Tensorflowは、公式のObject Detection API here をリリースしました。これは、たとえば、さまざまなスリム models で使用できます。

このAPIには、一般的なFaster RCNNを含むオブジェクト検出用のさまざまなパイプラインの実装と、事前にトレーニングされたモデルが含まれています。

14
gdelab