2017-10-17 16 views
0

私はテンソルフローで画像処理ネットワークを構築しており、テクスチャ損失を利用したいと考えています。事前に読み込まれたモデルが読み込まれていれば、テクスチャの損失は簡単に実装できます。テンソルフローのKerasモデル

TFを使用して私のモデルの計算グラフを作成しています。Keras.application.VGG19モデルを組み込み、レイヤ 'block4_conv4'からの出力を取得します。

問題があります:私は私のメインモデルから2つのTFテンソルの目標と結果を持っています。同じセッションでkeras VGG19にフィードしてその差分を計算し、私のモデルの主損失に使う方法はありますか?

答えて

0

次のコードは、右の作成セッションの後、私は重みに を開始することphi_func.load_weights(path)をトリック

with tf.variable_scope("") as scope: 
    phi_func = VGG19(include_top=False, weights=None, input_shape=(128, 128, 3)) 

    text_1 = phi_func(predicted) 
    scope.reuse_variables() 
    text_2 = phi_func(x) 

    text_loss = tf.reduce_mean((text_1 - text_2)**2) 
  • を呼ぶんようです
関連する問題