2017-01-18 9 views
1

既存のトレーニング済みmxnetモデルavailable hereのいずれかを完全畳み込み型に変換したいと考えています。事前トレーニングされたmxnetネットワークを完全に畳み込みにする

これは、任意のサイズの画像を入力し、ストライドを指定し、フル出力を得ることを意味します。 たとえば、モデルが224x224x3イメージでトレーニングされているとします。私は226x226x3の画像を入力し、3x3xnum-class出力を得るためにstride = 1を指定したいと思います。 私は「理論的に」求めているのではなく、コード例として:-)

ありがとう!この例によれば

答えて

1

https://github.com/dmlc/mxnet-notebooks/blob/master/python/tutorials/predict_imagenet.ipynb

あなたがモデルバインディング時にデータの形状を変更することができます。

mod.bind(for_training=False, data_shapes=[('data', (1,3,226,226))]) 

を次にあなたが入力3 * 226 * 226の画像をすることができます。

もう一つの例:http://mxnet.io/how_to/finetune.html

この例では、FC層との事前訓練を受けたモデルの最後の層を置き換えます。

+0

答えをありがとう。実際、入力サイズを変更できます。しかし、これはネットワークがいくつかの場所で網羅的に適用されることにはなりません。出力は同じサイズのままです。 – zuuz

+0

出力を変更する場合は、出力レイヤーの変更を検討することができます。 http://mxnet.io/how_to/finetune.html – kevinthesun

関連する問題