def _map_func(img, label):
    img = tf.image.resize_images(img, [img_shape[0], img_shape[1]], method=tf.image.ResizeMethod.BICUBIC)
    img = (tf.clip_by_value(tf.cast(img, tf.float32), 0, 255) / 255)
    return (img, label)
def _map_func(img, label):
    img = tf.image.resize_images(img, [img_shape[0], img_shape[1]], method=tf.image.ResizeMethod.BICUBIC)
    img = (tf.clip_by_value(tf.cast(img, tf.float32), 0, 255) / 255)
    return (img, label)