Keras搭建的卷積層為什麼會出現負尺寸?

在使用Keras進行模型搭建時,遇到了輸出為負尺寸的奇怪問題。

首先,這個模型我在其他電腦上是正常使用的,在一台新電腦上配置好keras環境後直接把代碼拷過來運行時遇到了錯誤。

錯誤如圖,列印出模型的結構時,發現輸出的尺寸參數含有負數!!!在原來的電腦上是不會出現這種情況的。

搭建模型的代碼如下圖:

這是什麼詭異的情況,對於解決這個問題一點頭緒都沒有,請各位指教。


我是題主,我發現了自己的錯誤所在。在keras.json文件中修改成"image_dim_ordering": "th",原來默認的是「tf」,圖像色彩通道的參數位置跑到第3個參數去了,所以和我的輸入尺寸不一致,才導致了這個問題。


Tensorflow和Theano的圖像數據參數排列順序有差異,這點似乎在中文文檔有提到:

一些基本概念 - Keras中文文檔

其中「data_format」中有提到


我的經驗:

如果您在windows下python環境的話,您把

input_shape=(1,64,64) 改成input_shape=(64,64,1) ,應該就可以


推薦閱讀:

深度學習伺服器配置(4-5w預算)?
FancyKeras-數據的輸入(花式)
一次關於嚴重侵犯著作權的事件
Keras,Theano,pytorch,caffe2 哪個更好一些,應該怎麼嘗試學習?
初探Keras&神經網路

TAG:深度學習DeepLearning | Theano | Keras |