在CNN
层之后,我使用Lambda
函数将每个过滤器(在CNN
上创建的图像)一个接一个地提供给LSTM
。
现在,我想改变这一点,使LSTM与所有过滤器一起水平连接在一起。
我看到了一些水平连接图像的代码,如下所示:
images = encoded_imgs[0,:,:,:] // eg: first output of a CNN layer
print(images[1,1,:].size)
image=images[:,:,0]
for i in range(1,10):
image=hstack((image,images[:,:,i]))
但是,如何将其应用于lambda函数以更改我的示例以将所有内容输入一个LSTM
发布于 2018-02-16 11:16:14
好的,我设法做到这一点,在lambda函数之后使用合并连接轴2来“个性化”每一个“过滤图像”。
conv2=Conv2D(filters,(3,3), padding="same")(conv1)
conv2=AveragePooling2D(pool_size=(2, 1), strides=(1, 1))(conv2)
conv2=Activation("relu")(conv2)
channels=Dropout(0.40)(conv2)
filtersVec=[]
for x in range(0,filters):
filterImg=Lambda(lambda element : element[:,x,:,:])(channels)
filtersVec.append(filterImg)
merged = merge(filtersVec, mode='concat',concat_axis=2)
lstm=Bidirectional(LSTM(20),merge_mode='concat',dropout=0.35)(merged)
classificationLayer=Dense(classes)(lstm)
classificationLayer=Activation("softmax")(classificationLayer)
https://stackoverflow.com/questions/48831893
复制