問題タブ [pruning]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
tensorflow - Tensorflowは事前訓練されたモデルからレイヤーを削除します
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 コードにアクセスできません。
ありがとう。