簡體   English   中英

我會嘗試為 keras 預訓練 model 更改頻道

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM