Wie benutz_ops in Tensorflow Android APP

Question

Ich möchte Transplantation schneller RCNN auf Android-Gerät, und ich treffe einige Probleme. Ich baue mein Android-Projekt von Bazel, genau wie die Tensorflow-Demo. Ich füge die op roi_pooling_op.cc und proposal_op.cc zu user_ops, aber ich scheint Android tut user_ops bauen, zeigt die logcat, wie folgend:

tensorflow_inference_jni.cc:146 Could not create TensorFlow graph: Not found: Op type not registered 'RoiPooling' 

Ich versuche, dieses Problem zu lösen, und ich bewege roi_pooling_op.cc und proposal_op.cc zu jni Ordner (vielleicht ist es ein schlechtes ideal) ist, wird der vorherige Fehler verschwunden, aber ich treffe ein anderes Problem, Ausgänge gefunden alle nicht, ich weiß nicht, warum, logcat wie folgend:

tensorflow_inference_jni.cc:170 Output [rois] not found, aborting! 
tensorflow_inference_jni.cc:170 Output [bbox_pred/bbox_pred] not found, aborting! 
tensorflow_inference_jni.cc:170 Output [cls_prob] not found, aborting! 

ich weiß nicht, wie sich zu lösen und Ich debugge, könntest du mir helfen.

Vielen Dank im Voraus!

Answer 1

Gibt es vielleicht einen Ausgabe-Index für diese Knoten? Z.B. "rois: 0" oder "rois: 1".

Wenn Sie im Konstruktor eine Protokollzeile für Knotennamen/-typen zu stat_summarizer.cc hinzufügen, können Sie sehen, welche Knoten in Ihrem Diagramm definiert sind, um das Problem zu verdeutlichen.

Das Hinzufügen der Ops zum jni-Verzeichnis, wie Sie es getan haben, sollte als eine schnelle und schmutzige Lösung gut funktionieren, und es scheint so zu sein, wenn es Ihr anfängliches Problem überstanden hat.