小编Meg*_*ude的帖子

如何拆分卷积自编码器?

我已经编译了一个自动编码器(完整代码如下),在训练后我想将它分成两个单独的模型:编码器(层 e1...编码)和解码器(所有其他层),在其中输入手动修改的图像由解码器编码的。我已经成功地将编码器创建为一个单独的模型:

encoder = Model(input_img, autoencoder.layers[6].output)
Run Code Online (Sandbox Code Playgroud)

但是当我尝试制作解码器时,同样的方法失败了:

encoded_input = Input(shape=(4,4,8))
decoder = Model(input_img, decoded)
Run Code Online (Sandbox Code Playgroud)

这是我的完整代码:

from keras.layers import Input, Dense, Conv2D, MaxPooling2D, UpSampling2D
from keras.models import Model
from keras import backend as K
from keras.models import load_model

input_img = Input(shape=(28, 28, 1))  # adapt this if using channels_first` image data format

e1 = Conv2D(16, (3, 3), activation='relu', padding='same')(input_img)
e2 = MaxPooling2D((2, 2), padding='same')(e1)
e3 = Conv2D(8, (3, 3), activation='relu', padding='same')(e2)
e4 = MaxPooling2D((2, 2), padding='same')(e3)
e5 = Conv2D(8, (3, 3), …
Run Code Online (Sandbox Code Playgroud)

autoencoder keras

7
推荐指数
1
解决办法
2215
查看次数

标签 统计

autoencoder ×1

keras ×1