Keras 如何修改图片通道的顺序
1. error
在使用keras.layers.convolutional.convolution2d的时候
convolution2d(32, 5, 5, border_mode='valid', input_shape=(1, 28, 28), activation='relu')
报如下错误:
overflowerror: range exceeds valid bounds
这是因为keras配置图片通道顺序错误导致的。上面代码使用图片通道顺序是[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)
所以,要保证你使用的通道顺序和配置的通道顺序一致
或者通过这样修改
from keras import backend backend.set_image_dim_ordering('th')
补充:keras设置后端数据维度顺序
keras是比较高级的深度学习框架,其对几个比较常见的主流深度学习框架的封装而来,也就是按照keras的说法就是其后端是以tensorflow、theano和cntk为基础封装而来。
cntk我没做过多了解,tensorflow的tensor的维度顺序和theano的维度顺序是不一样的,所以在使用keras的时候要格外注意这一点,如果后端不同的话,设置的数据维度顺序也是要不一样的。
tensorflow的数据维度默认顺序是channels_last的好像,也就是说其一个tensor的数据维度是[samples,rows,cols,channels],而theano不同,其数据顺序是channels_first,也就是[samples,rows,cols,channels],这个其实也就是这两个框架的不同而已,如果用keras来写深度学习模型的话,其实就只要设置好后端基本也没啥问题,也不必太考虑数据维度的问题。
但是keras的数据维度是可以改变的,即使是使用tensorflow为后端,keras的数据格式也可以是channels_first的,并且这个是可以通过配置文件来改,也可以通过代码来改的。
keras的配置文件默认是在$home/.keras/keras.json,可以用txt编辑器打开直接修改后端配置的:
{ "floatx": "float32", "epsilon": 1e-07, "image_data_format": "channels_last", "backend": "tensorflow" }
这是我的windows下的配置信息,默认是以tensorflow为后端,数据格式是channels_last,但是我有个小项目是一开始是以theano为后端来写的,本来就只要稍微改一下就好,但是因为只是这个项目需要改为theano,所以我并没有直接改配置文件,而是在代码里修改后端,这里需要用到keras的后端backend来实现,修改的代码如下:
from keras import backend as bk bk.set_image_data_format("channels_first") bk.set_image_dim_ordering("th")
这样设置之后再这个程序跑起来的时候其后端并不会被改变,改变的只是数据的维度顺序而已,而且仅仅是针对当前的程序,对其它程序是不影响的。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持。
下一篇: python爬虫之异常捕获及标签过滤详解