TrickyGo/Dive-into-DL-TensorFlow2.0

3.5 读取小批量 像素标准化

taichuai opened this issue · 6 comments

3.5 读取小批量
并除以255使得所有像素的数值均在0到1之间
应该加一个 map函数作映射
def data_trans(x, y):
x = tf.cast(x, tf.float32)
x = x / 255
return x, y

train_iter = tf.data.Dataset.from_tensor_slices((x_train, y_train)).map(data_trans).batch(256)

不理解为什么要加map函数做映射

不理解为什么要加map函数做映射

因为像素点的值是 0-255

这个是读取数据,为了展示作用,原书就是这样做的

不过除以255,图片展示效果应该不一样了