keras 2.3.0 做上采样 UpSampling2D的时候的维度出错问题解决办法
简单的说,你是不是遇到了这样的问题,上一层的数据是【None,200,14,14】
你希望上采样到28x28
H = UpSampling2D(size=(2, 2))(H)
你以为能得到【None,200,28,28】
结果却成了【None,400,28,14】
维度上的错位问题,需要一个参数来解决,我看别的帖子都没有,在官方文档里找到了解决办法。
UpSampling2D,有个参数叫data_format
data_format默认参数是'channels_last',默认最后的维数是ch
原创
2020-06-14 21:27:32 ·
1315 阅读 ·
0 评论