1. 程式人生 > 程式設計 >基於K.image_data_format() == 'channels_first' 的理解

基於K.image_data_format() == 'channels_first' 的理解

我們在學習keras經常會看到下面這樣的程式碼段:

基於K.image_data_format() == 'channels_first' 的理解

查閱官方文件可以知道:

基於K.image_data_format() == 'channels_first' 的理解

我們知道彩色影象一般會有Width,Height,Channels,而“channels_first”或“channels_last”,則代表資料的通道維的位置。

該引數是Keras 1.x中的image_dim_ordering,“channels_last”對應原本的“tf”,“channels_first”對應原本的“th”。

以128x128x128的資料為例,“channels_first”應將資料組織為(3,128,128),而“channels_last”應將資料組織為(128,3)。

該引數的預設值是~/.keras/keras.json中設定的值,若從未設定過,則為“channels_last”。

補充知識:channels_first 與 channels_last 的區別

用卷積神經網路處理一組彩色圖片時,Caffe/Theano 使用的資料格式是channels_first即:

(樣本數,通道數,行數(高),列數(寬))

Tensforflow 使用的資料格式是channels_last即:

(樣本數,行數(高),列數(寬),通道數)

以上這篇基於K.image_data_format() == 'channels_first' 的理解就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。