ValueError: Depth of output (64)

2020-08-05  本文已影响0人  是我真的是我

上述是训练CNN网络时,在卷积层运算遇到的错误。

若要消除此错误,需要将该层卷积核深度(个数)改为3的倍数,同时注意与其他卷积层深度呈倍数关系(一般1~3倍)

至于原因,可参考如下:https://www.zhihu.com/question/263391986
一个原因是随着卷积神经网络的加深,一般会用更小的卷积核,再加上经过池化减小的计算量可以提供给增加的卷积核;
二是因为越深层的网络提取的特征越抽象,所以通过增加卷积核的数量将前面层学习到的特征组合起来,从而使得各种特征的可能性覆盖得更全面。

上一篇 下一篇

猜你喜欢

热点阅读