PyTorch的输出和keras的输出之间的不同

xiaoxiaokeke 2020-06-14

PyTorch的输入输出格式为:                          keras的输入输出格式为: 

input(N,C-in,H-in,W-in)                                      input(N,H-in,W-in,C-in)

output(N,C-out,H-out,W-out)                             output(N,H-out,W-out,C-out)

其中:N是batch的大小,C是通道数量,H是输入的高度,W是输入的宽度

在进行特征图拼接时特别要注意这一点,注意用的是什么框架。

相关推荐