1

Tensorflow で事前トレーニング済みのモデルをロードし、ネットワークの最上位レイヤーを削除する方法はありますか? Tensorflowリリースr1.10を見ています

私が見つけた唯一のドキュメントはtf.keras.Sequential.pop https://www.tensorflow.org/versions/r1.10/api_docs/python/tf/keras/Sequential#popにあります

一連の最上位の畳み込み層を削除し、カスタムの完全に畳み込み層を追加して、事前トレーニング済みのネットワークを手動でプルーニングしたいと考えています。

編集:

モデルは Tensorflow Model Zooからダウンロードした ssd_mobilenet_v1_cocoです。frozen_inference_graph.pb モデル ファイルとチェックポイント ファイルの両方にアクセスできます。

モデルの構築に使用される python コードにアクセスできません。

ありがとう。

4

1 に答える 1