[英]I would try change channel for keras pretrained model
我得到了一個 Xception model。
Xception = tf.keras.applications.Xception(input_shape=(512, 512, 3), include_top=False)
我結合了 model 將輸入通道更改為 3。
input_layer = keras.Input(shape=(512, 512, 1), name="img_input")
x = layers.UpSampling3D(size=(1, 1, 3), name="output")(input_layer)
input_model = keras.Model(input_layer, x, name="input_model")
model = keras.Model(input_model, Xception, name="model")
但是我有錯誤
Input tensors to a Functional must come from `tf.keras.Input`. Received: <tensorflow.python.keras.engine.functional.Functional object at 0x7f922942a690> (missing previous layer metadata).
您只需在新的 model 中以正確的方式嵌入Xception
:
Xception = tf.keras.applications.Xception(input_shape=(512, 512, 3), include_top=False)
input_layer = tf.keras.Input(shape=(512, 512, 1), name="img_input")
x = tf.keras.layers.UpSampling3D(size=(1, 1, 3), name="upsampling")(input_layer)
output = Xception(x)
model = tf.keras.Model(input_layer, output, name="model")
我們創建一個新的Input
層,然后我們進行上采樣,最后我們將所有的都傳遞給Xception
如果您有興趣, 這是正在運行的筆記本
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.