发新帖

keras区分 channels_first (3,128,128) 和channels_last(128, 128...

[复制链接]
893 2

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
请教:keras区分 channels_first (3,128,128) 和channels_last(128, 128,3)的目的是什么?
tf或者keras中设置input_shape时不是很理解,通常希望我的输入是向量时,那么input_shape=n_features就行了,但往往需要改变维度。
舟3332已获得悬赏 10 金币+5 金币

最佳答案

今天刚好看到了这里。 channel_first 也叫 NCHW channel_last 也叫 NHWC 具体选择哪一个是和你的训练环境有关。一般来说在 CPU 上 channel_first 会快一点。GPU 上相反。这篇文章讲的比较好,推荐一下~ https:// ...
本楼点评(0) 收起

精彩评论2

舟3332  TF芽芽  发表于 2018-4-23 20:32:52 | 显示全部楼层
今天刚好看到了这里。

channel_first 也叫 NCHW
channel_last 也叫 NHWC

具体选择哪一个是和你的训练环境有关。一般来说在 CPU 上 channel_first 会快一点。GPU 上相反。这篇文章讲的比较好,推荐一下~

https://mp.weixin.qq.com/s/I4Q1Bv7yecqYXUra49o7tw
本楼点评(0) 收起
neverchange  TF豆豆  发表于 2018-7-3 23:39:10 | 显示全部楼层
可以相互转换
NHWC [batch, in_height, in_width, in_channels]
NCHW [batch, in_channels, in_height, in_width]
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

14

积分
快速回复 返回顶部 返回列表