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

Posted 2020-12-10 mydream6

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是输入的宽度

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