Tensorflowモデルをオブジェクト検出APIからuffに変換

Question

ボックスMobilenet（およびTF OD APIの他のモデル）からuffフォーマットに変換してからTensorRTフォーマットbecause of much unsupported layersに変換する方法がないようです。 このレイヤーを削除する方法はありますか？たとえば、graph_transformツールを使用していますか？私はこれらの層のすべてではないという目的を理解しています。 誰かが試してみたい場合、ここにはdefault modelがあります。

Answer 1

UFF変換ツールは、一部の認識されないレイヤーでチョークしているようですが、これはGAリリースで改善される可能性があります。今のところ、これらのレイヤーを削除し（推論に必要な最小のサブセットのみを保持する）、nvinferプラグインAPIを使用してこれらのレイヤーを実装する必要があります。