def parse(mode, image, label):
"""Parse input record to features and labels."""
image = tf.to_float(image)
label = tf.to_int64(label)
image = tf.image.per_image_standardization(image)
return {"image": image}, {"label": label}
评论列表
文章目录