2016-07-13 43 views
40

有沒有人爲TensorFlow版本實現FRCNN? 我發現了一些相關的回購協議如下:TensorFlow更快的RCNN

  1. Implement roi pool layer
  2. Implement fast RCNN based on py-faster-rcnn repo

,但1:假設投資回報率彙集層工作(我沒試過),有什麼需要實施如下:

  • ROI數據層例如roidb
  • 線性迴歸例如SmoothL1Loss
  • 用於端到端培訓的ROI池層後期處理,應將ROI池化層的結果轉換爲分類器的CNN結果。

對於2:em ....似乎基於py-faster-rcnn,它基於Caffe準備預處理(例如roidb)並將數據饋入Tensorflow以訓練模型,它似乎很奇怪,所以我可能沒有嘗試過。我想知道的是,Tensorflow support Faster RCNN in the future?。如果沒有,我是否有任何上面提到的錯誤理解?或有任何回購或某人的支持?

+0

SmoothL1Loss應該相對容易實現使用ROI池的實際tf不知道... – jean

+0

我正在處理您的問題的類似目標。我發現很難用張量表示動態bbox。這也許就是你提到的方法2使用caffe來預處理數據的原因。我試圖找出在TensorFlow中是否有其他方法來實現這一點。 –

+4

[此實現]如何(https://github.com/smallcorgi/Faster-RCNN_TF)? – Shai

回答

10

Tensorflow剛剛發佈了一個官方的對象檢測API here,它可以用於他們的各種苗條models

該API包含各種管道對象檢測的實現,包括受歡迎的快速RCNN以及其預先訓練的模型。

+6

所有mod/reviewers:請做**不**刪除這個答案爲「僅鏈接」:問題的性質允許這樣的答案,這個特定的答案是好的。 – Shai

+1

鏈接似乎不存在 – zavtra

+2

我修復了它們 – gdelab