問題タブ [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.

0 投票する
1 に答える
2765 参照

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 コードにアクセスできません。

ありがとう。