Keras怎么修改圖片通道的順序

這篇文章主要介紹了Keras怎么修改圖片通道的順序，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

1. Error

在使用Keras.layers.convolutional.Convolution2D的時(shí)候

Convolution2D(32, 5, 5, border_mode='valid', input_shape=(1, 28, 28), activation='relu')

報(bào)如下錯(cuò)誤：

OverflowError: Range exceeds valid bounds

這是因?yàn)镵eras配置圖片通道順序錯(cuò)誤導(dǎo)致的。上面代碼使用圖片通道順序是[channels][height][width]

2. 修改圖片通道順序

檢查 ~/.keras/keras.json文件

if "image_dim_ordering": is "th" and "backend": "theano", your input_shape must be (channels, height, width)
if "image_dim_ordering": is "tf" and "backend": "tensorflow", your input_shape must be (height, width, channels)

所以，要保證你使用的通道順序和配置的通道順序一致

或者通過(guò)這樣修改

from keras import backend
backend.set_image_dim_ordering('th')

補(bǔ)充：Keras設(shè)置后端數(shù)據(jù)維度順序

Keras是比較高級(jí)的深度學(xué)習(xí)框架，其對(duì)幾個(gè)比較常見(jiàn)的主流深度學(xué)習(xí)框架的封裝而來(lái)，也就是按照Keras的說(shuō)法就是其后端是以Tensorflow、Theano和CNTK為基礎(chǔ)封裝而來(lái)。

CNTK我沒(méi)做過(guò)多了解，tensorflow的tensor的維度順序和Theano的維度順序是不一樣的，所以在使用Keras的時(shí)候要格外注意這一點(diǎn)，如果后端不同的話，設(shè)置的數(shù)據(jù)維度順序也是要不一樣的。

tensorflow的數(shù)據(jù)維度默認(rèn)順序是channels_last的好像，也就是說(shuō)其一個(gè)tensor的數(shù)據(jù)維度是[samples，rows，cols，channels]，而Theano不同，其數(shù)據(jù)順序是channels_first，也就是[samples，rows，cols，channels]，這個(gè)其實(shí)也就是這兩個(gè)框架的不同而已，如果用keras來(lái)寫(xiě)深度學(xué)習(xí)模型的話，其實(shí)就只要設(shè)置好后端基本也沒(méi)啥問(wèn)題，也不必太考慮數(shù)據(jù)維度的問(wèn)題。

但是Keras的數(shù)據(jù)維度是可以改變的，即使是使用tensorflow為后端，Keras的數(shù)據(jù)格式也可以是channels_first的，并且這個(gè)是可以通過(guò)配置文件來(lái)改，也可以通過(guò)代碼來(lái)改的。

Keras的配置文件默認(rèn)是在$HOME/.keras/keras.json，可以用txt編輯器打開(kāi)直接修改后端配置的：

{
    "floatx": "float32",
    "epsilon": 1e-07,
    "image_data_format": "channels_last",
    "backend": "tensorflow"
}

這是我的windows下的配置信息，默認(rèn)是以tensorflow為后端，數(shù)據(jù)格式是channels_last，但是我有個(gè)小項(xiàng)目是一開(kāi)始是以Theano為后端來(lái)寫(xiě)的，本來(lái)就只要稍微改一下就好，但是因?yàn)橹皇沁@個(gè)項(xiàng)目需要改為T(mén)heano，所以我并沒(méi)有直接改配置文件，而是在代碼里修改后端，這里需要用到Keras的后端backend來(lái)實(shí)現(xiàn)，修改的代碼如下：

from keras import backend as BK
BK.set_image_data_format("channels_first")
BK.set_image_dim_ordering("th")

這樣設(shè)置之后再這個(gè)程序跑起來(lái)的時(shí)候其后端并不會(huì)被改變，改變的只是數(shù)據(jù)的維度順序而已，而且僅僅是針對(duì)當(dāng)前的程序，對(duì)其它程序是不影響的。

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“Keras怎么修改圖片通道的順序”這篇文章對(duì)大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來(lái)學(xué)習(xí)!