Breaking News: Grepper is joining You.com. Read the official announcement!

images to tf.dataset.Dataset

VVY answered on April 10, 2023 Popularity 3/10 Helpfulness 8/10

answer images to tf.dataset.Dataset

images to tf.dataset.Dataset

Comment

# step 1
filenames = tf.constant(["img1.jpg", ...])
labels = tf.constant([1, ...])

# step 2: create a dataset returning slices of `filenames`
dataset = tf.data.Dataset.from_tensor_slices((filenames, labels))

# step 3: parse every image in the dataset using `map`
def _parse_function(filename, label):
    image_string = tf.io.read_file(filename)
    tensor = tf.io.decode_jpeg(image_string, channels=3)
    tensor = tf.image.resize(tensor, [224,224])
    image = tf.cast(tensor, tf.float32)/255. # Normalizing
    return image, label

dataset = dataset.map(_parse_function).batch(32)

Popularity 3/10 Helpfulness 8/10 Language python

Source: stackoverflow.com

Tags: image path python

Link to this answer
Share Copy Link