tensorflow遇到的坑

Author Avatar
Patrick 3月 09, 2020

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 转化为浮点型,这时候就只转换数据类型不改变值的大小了。