2017-10-17 93 views
0

我在張量流中構建圖像處理網絡,我想利用紋理損失。如果你已經預裝了模型,紋理損失似乎很容易實現。張量流中的Keras模型

我使用TF爲我的模型構建計算圖,並且我想要合併Keras.application.VGG19模型以從圖層'block4_conv4'獲取輸出。

問題是:我有兩個TF張量目標和結果從我的主模型,如何將它們送入keras VGG19在同一個會話中計算它們的差異並將其用於我的模型的主要損失?

回答

0

它似乎下面的代碼做的伎倆

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) 
  • 之後會話創建我打電話phi_func.load_weights(path)啓動權重