tensorflow遇到的坑 | Patrick's Space

IO 的坑

对于tf.io.decode_image()，如果指定 dtype=tf.float32，那么会默认把 tensor 的值除以 255。

我之前就在这里直接指定了，后面还减掉了 ImageNet 的均值（[123.68, 116.779, 103.939]），所以对于任何的图，其 RGB 通道上的值都基本上是 [-123.68, -116.779, -103.939]，也难怪网络的预测结果都是数量最多的那两个标签……

如果要像我这么做的话，就不指定 dtype，然后再用 tf.cast() 将 tensor 转化为浮点型，这时候就只转换数据类型不改变值的大小了。