ubuntu下keras的卷积padding错位(pad在channel上)

修改keras的后端。

方法参考Keras中文文档:http://keras-cn.readthedocs.io/en/latest/backend/

切换后端

注意:Windows用户请把$Home改为%USERPROFILE%

如果你至少运行过一次Keras,你将在下面的目录下找到Keras的配置文件:

$HOME/.keras/keras.json

如果该目录下没有该文件,你可以手动创建一个

文件的默认配置如下:

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

backend字段的值改写为你需要使用的后端:theanotensorflow或者CNTK,即可完成后端的切换

我们也可以通过定义环境变量KERAS_BACKEND来覆盖上面配置文件中定义的后端:

KERAS_BACKEND=tensorflow python -c "from keras import backend;"
Using TensorFlow backend.

keras.json 细节

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

你可以更改以上~/.keras/keras.json中的配置

扫描二维码关注公众号,回复: 1927844 查看本文章
  • iamge_data_format:字符串,"channels_last"或"channels_first",该选项指定了Keras将要使用的维度顺序,可通过keras.backend.image_data_format()来获取当前的维度顺序。对2D数据来说,"channels_last"假定维度顺序为(rows,cols,channels)而"channels_first"假定维度顺序为(channels, rows, cols)。对3D数据而言,"channels_last"假定(conv_dim1, conv_dim2, conv_dim3, channels),"channels_first"则是(channels, conv_dim1, conv_dim2, conv_dim3)

  • epsilon:浮点数,防止除0错误的小数字

  • floatx:字符串,"float16", "float32", "float64"之一,为浮点数精度
  • backend:字符串,所使用的后端,为"tensorflow"或"theano"


猜你喜欢

转载自blog.csdn.net/qq_33266320/article/details/80067082