如何创建自动编码器的解码器部分?

2024-09-29 01:23:14 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图从编码器复制层来创建解码器,但我得到了“索引错误”

input_img =Input(25425,)

encoded1 = Dense(75,activation=tf.nn.relu)(input_img)

encoded = Dense(50,activation=tf.nn.relu)(encoded1)

decoded = Dense(25425, activation='sigmoid')(encoded)

autoencoder = Model(input_img, encoded1, decoded)

encoder = Model(input_img, encoded)

encoded_input = Input(shape=(50,))


decoder_layer1 = autoencoder.layers[1](encoded_input)

decoder_layer2 = autoencoder.layers[0](decoder_layer1)

decoder = Model(encoded_input, decoder_layer1, decoder_layer2)

autoencoder.compile(optimizer='adadelta', loss='binary_crossentropy')

autoencoder.fit(X, X,
            epochs=50,
            shuffle=True)

我希望解码器有相同的层和编码器刚刚颠倒,但我不能复制层以上。我得到这个错误:

Traceback (most recent call last):
File "C:\Users\dalto\Documents\geo4\train.py", line 36, in <module>
decoder_layer1 = autoencoder.layers[1](encoded_input)
IndexError: list index out of range

Tags: imginputmodellayers错误编码器解码器activation
1条回答
网友
1楼 · 发布于 2024-09-29 01:23:14

你的代码有几个错误。在工作片段中查看我的评论:

# Random input for testing purposes
X = np.random.rand(10, 25425)

input_img =tf.keras.layers.Input(25425,)
encoded1 = tf.keras.layers.Dense(75,activation=tf.nn.relu)(input_img)
encoded2 = tf.keras.layers.Dense(50,activation=tf.nn.relu)(encoded1)
decoded = tf.keras.layers.Dense(25425, activation='sigmoid')(encoded2)
# The input of the autoencoder is the image (input_img), and the output is the decoder layer (decoded)
autoencoder = tf.keras.Model(input_img, decoded)

encoder = tf.keras.Model(input_img, encoded2)

encoded_input = tf.keras.layers.Input(shape=(50,))
# The decoded only consists of the last layer
decoder_layer = autoencoder.layers[-1](encoded_input)
# The input to the decoder is the vector of the encoder which will be fed (using encoded_input), the output is the last layer of the network (decoder_layer)
decoder = tf.keras.Model(encoded_input, decoder_layer)

autoencoder.compile(optimizer='adadelta', loss='binary_crossentropy')
autoencoder.fit(X, X, epochs=50, shuffle=True)

相关问题 更多 >