在keras中獲取某一層上的feature map實例
在深度學習中,如果我們想獲得某一個層上的feature map,就像下面的圖這樣,怎么做呢?
我們的代碼是使用keras寫的VGG16網絡,網絡結構如圖:
那么我們隨便抽取一層的數據吧,就拿第四層的pooling以后的結果作為輸出吧,參考上面的網絡結構,得到的結果維度應該是[1,56,56,128]的尺度。
怎么做呢?
首先通過keras構建模型:
model = VGG16(include_top=True, weights='imagenet')
然后設置輸入和輸出為:原始的輸入和該層對應的輸出,然后使用predict函數得到對應的結果
dense_result = Model(inputs=model.input,outputs=model.get_layer("block2_pool").output) dense_res = dense_result.predict(x)#使用predict得到該層結果
設置隨機數(或者固定的數字)來獲取某一層的結果
rand_layer = random.randint(10,128) x_output = dense_res[0,:,:,rand_layer] #獲取某一層的數據:因為原始數據維度是[1,x,x,depths]的,我們僅僅提取某一個depth對應的[x,x]維度的信息 # 獲取最大值,然后對該層數據進行歸一化之后投影到0-255之間 max = np.max(x_output) print(max,"max value is :") # 然后進行歸一化操作 x_output =x_output.astype("float32") / max * 255 print(x_output.shape)
最后對該層的feature進行顯示,我們使用Pillow庫
# 把圖像轉換成image可以表示的方式進行顯示 from PIL import Image as PILImage x_output =PILImage.fromarray(np.asarray(x_output)) x_output1 = x_output.resize((400,400)) x_output1.show() print(np.asarray(x_output1))
結果如上圖所示啦~
以上這篇在keras中獲取某一層上的feature map實例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Python實現讀取.nc數據并提取指定時間與經緯度維度對應的變量數值
這篇文章主要為大家詳細介紹了如何使用Python語言的netCDF4庫實現讀取.nc格式的數據文件,并提取指定維(時間、經度與緯度)下的變量數據,需要的可以了解下2024-02-02Python庫urllib與urllib2主要區(qū)別分析
這篇文章主要介紹了Python庫urllib與urllib2主要區(qū)別,需要的朋友可以參考下2014-07-07