Keras Conv2D和Pytorch Conv2D图片格式的区别

Conv2D中Keras默认input格式和Pytorch input格式的差异

  • Keras的Conv2D的默认输入格式是 batch, width, height, channel
  • Pytorch的Conv2D是 batch, channel, width, height

下面是图例

keras:

keras官方文档的输入格式,注意看到data_format这个属性,修改data_format为"channels_first"和"channels_last"会影响可以接受的input的形状

Keras Conv2D和Pytorch Conv2D图片格式的区别_第1张图片
data_format官方文档

由上图可见默认是batch, width, height, channels

Keras Conv2D和Pytorch Conv2D图片格式的区别_第2张图片
代码实例

Keras Conv2D和Pytorch Conv2D图片格式的区别_第3张图片

pytorch

Keras Conv2D和Pytorch Conv2D图片格式的区别_第4张图片

杂记

注意图片的width,宽,是看横向距离
图片的height,高,是看纵向距离

你可能感兴趣的:(深度学习)