keras 使用Lambda 快速新建層 添加多個參數(shù)操作
keras許多簡單操作,都需要新建一個層,使用Lambda可以很好完成需求。
# 額外參數(shù)
def normal_reshape(x, shape):
return K.reshape(x,shape)
output = Lambda(normal_reshape, arguments={'shape':(-1, image_seq, 1000)})(output)
output = Lambda(lambda inp: K.mean(inp, axis=1), output_shape=(1000,))(output)
補(bǔ)充知識:keras 實現(xiàn)包括batch size所在維度的reshape,使用backend新建一層 針對多輸入使用不同batch size折衷解決辦法
新建層,可以在此層內(nèi)使用backend完成想要的功能,如包含batch size維度在內(nèi)的reshpe:
def backend_reshape(x): return backend.reshape(x, (-1, 5, 256))
使用lambda方法調(diào)用層:
vision_model.add(Lambda(backend_reshape, output_shape=(5, 256)))
注意指定輸出維度
在多輸入問題中,有時兩個輸入具有不同的batch size,但在keras無法直接實現(xiàn)。我所遇到的問題是,我有兩個輸入分別是圖像輸入和問題輸入,對于圖像輸入每個樣本是一個圖像序列。這就要求我們在把圖像序列輸入到CNN中時是一張一張圖像。
我的解決辦法是在輸入是把圖像序列作為一個樣本,等輸入進(jìn)去后,通過上述的reshape方法將圖像序列重新拆分成一張張圖像輸入到CNN,然后在后期處理時重新reshape成一個序列樣本。
代碼:
image_seq = 4 def preprocess_reshape(x): return K.reshape(x, (-1, 224, 224,3)) def backend_reshape(x): return K.reshape(x, (-1, image_seq, 256))
image_input = Input(shape=(image_seq, 224, 224, 3) , name='input_img') image_re = Lambda(preprocess_reshape, output_shape=(224,224,3))(image_input) im_pre = Lambda(preprocess_input, name='preprocessing')(image_re)
vision_model.add(Lambda(backend_reshape, output_shape=(image_seq, 256))) vision_model.add(LSTM(256, kernel_regularizer=l2, recurrent_regularizer=l2))
以上這篇keras 使用Lambda 快速新建層 添加多個參數(shù)操作就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
numpy數(shù)組的重塑和轉(zhuǎn)置實現(xiàn)
本文主要介紹了numpy數(shù)組的重塑和轉(zhuǎn)置實現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
Python網(wǎng)絡(luò)爬蟲之Web網(wǎng)頁基礎(chǔ)
我們在學(xué)習(xí)爬蟲之前,要先了解網(wǎng)頁的組成,只有我們了解其組成嗎,才可以方能百戰(zhàn)百勝,文章中有詳細(xì)的代碼示例,需要的朋友可以參考一下2023-04-04
Python 動態(tài)變量名定義與調(diào)用方法
這篇文章主要介紹了Python 動態(tài)變量名定義與調(diào)用方法,需要的朋友可以參考下2020-02-02
詳解如何利用Pytest?Cache?Fixture實現(xiàn)測試結(jié)果緩存
這篇文章主要為大家詳細(xì)介紹了如何利用Pytest?Cache?Fixture實現(xiàn)測試結(jié)果緩存,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起了解一下2023-09-09

