2016-07-13 22 views
40

Hat jemand die FRCNN für TensorFlow Version implementieren? fand ich einige verwandte repos, wie folgend:Schneller RCNN für TensorFlow

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

aber für 1: die Schicht Werke roi Pooling nehmen (ich habe nicht versucht), und es gibt etwas brauchen, um wie folgt implementiert werden:

  • ROI Datenschicht z roidb.
  • Lineare Regression, z.B. SmoothL1Loss
  • Nachverarbeitung der ROI-Pool-Schicht für End-to-End-Training, die die Ergebnisse der ROI-Pooling-Schicht in CNN für Klassifikator konvertieren soll.

für 2: em .... scheint es, basierend auf py-schneller-RCNN, die bereit Vorbearbeitung auf Caffe basiert (zB roidb) und Feed-Daten in Tensorflow das Modell zu trainieren, wie es scheint seltsam Also kann ich es nicht versuchen.

Also was ich wissen möchte ist, dass Tensorflow support Faster RCNN in the future?. Wenn nicht, habe ich etwas falsch verstanden, was oben erwähnt wurde? oder hat ein Repo oder jemand das unterstützt?

+0

SmoothL1Loss sollte relativ einfach zu implementieren mit dem tatsächlichen tf für ROI-Pooling sein keine Ahnung ... – jean

+0

Ich arbeite an dem ähnlichen Ziel Ihrer Frage. Ich fand es schwierig, dynamische Bboxen im Tensor darzustellen. Das ist vielleicht der Grund, warum die von Ihnen erwähnte Methode 2 caffe zur Vorverarbeitung von Daten verwendet. Ich versuche herauszufinden, ob es in TensorFlow noch einen anderen Weg gibt. –

+4

Wie wäre es mit [dieser Implementierung] (https://github.com/smallcorgi/Faster-RCNN_TF)? – Shai

Antwort

10

Tensorflow hat soeben eine offizielle Object Detection API here, veröffentlicht, die zum Beispiel mit ihren verschiedenen schlanken models verwendet werden kann.

Diese API enthält die Implementierung verschiedener Pipelines zur Objekterkennung, einschließlich der beliebten Faster RCNN, mit ihren vortrainierten Modellen.

+6

an alle Mods/Rezensenten: bitte lösche ** NICHT ** diese Antwort als "nur Link": die Art der Frage erlaubt solche Antworten und diese ist in Ordnung. – Shai

+1

Links scheinen tot – zavtra

+2

Ich habe sie behoben – gdelab