Why Dataset from files in TensorFlow? - Purpose & Use Cases

The Big Idea

What if you could turn a mountain of files into a ready-to-use dataset with just a few lines of code?

The Scenario

Imagine you have hundreds or thousands of images or text files stored on your computer. You want to analyze or train a model using this data. Opening each file one by one, reading its content, and organizing it manually feels like sorting a huge pile of papers by hand.

The Problem

Manually loading files is slow and tiring. It's easy to make mistakes like missing files, mixing up data order, or running out of memory. Also, doing this repeatedly wastes time and energy that could be used for learning or improving your model.

The Solution

Using TensorFlow's Dataset from files lets you automatically load, shuffle, and batch your data efficiently. It handles large files smoothly and prepares your data step-by-step for training, so you don't have to worry about the messy details.

Before vs After

✗ Before

files = ['img1.jpg', 'img2.jpg']
data = []
for f in files:
    with open(f, 'rb') as file:
        data.append(file.read())

✓ After

dataset = tf.data.Dataset.list_files('images/*.jpg')
dataset = dataset.map(load_and_preprocess_image)

What It Enables

You can easily build powerful models that learn from huge collections of files without getting stuck on loading or organizing data.

Real Life Example

A data scientist training a cat vs. dog image classifier can load thousands of photos from folders automatically, shuffle them, and feed them into the model in batches, all with just a few lines of code.

Key Takeaways

Manually loading files is slow and error-prone.

TensorFlow Dataset from files automates and speeds up data loading.

This makes training on large file collections easy and efficient.

Practice

1. What is the main purpose of using tf.data.Dataset.from_tensor_slices() with file paths in TensorFlow?

easy

A. To convert tensors into image files

B. To directly read image data from files into memory

C. To save datasets to disk as files

D. To create a dataset that holds file paths which can be read later

Solution

1. Step 1: Understand the function purpose

   tf.data.Dataset.from_tensor_slices() creates a dataset from a tensor, often a list of file paths, not the file contents themselves.

2. Step 2: Clarify dataset content

   The dataset holds file paths as strings, which can be mapped later to read actual file data.

3. Final Answer:

   To create a dataset that holds file paths which can be read later -> Option D

4. Quick Check:

   from_tensor_slices(file_paths) = dataset of paths [OK]

Hint: Remember: from_tensor_slices holds paths, not file data [OK]

Common Mistakes:

• Thinking it reads file contents immediately
• Confusing dataset creation with saving files
• Assuming it converts tensors to images

2. Which of the following is the correct way to create a dataset from a list of image file paths in TensorFlow?

easy

A. dataset = tf.data.Dataset.from_tensor_slices(image_paths)

B. dataset = tf.data.Dataset.read_files(image_paths)

C. dataset = tf.data.Dataset.load(image_paths)

D. dataset = tf.data.Dataset.create(image_paths)

Solution

1. Step 1: Recall correct TensorFlow method

   The method to create a dataset from a list of tensors (like file paths) is from_tensor_slices().

2. Step 2: Verify options

   Methods like tf.data.Dataset.load(), tf.data.Dataset.read_files(), and tf.data.Dataset.create() are not valid TensorFlow dataset creation methods.

3. Final Answer:

   dataset = tf.data.Dataset.from_tensor_slices(image_paths) -> Option A

4. Quick Check:

   Correct method is from_tensor_slices [OK]

Hint: Use from_tensor_slices for lists of file paths [OK]

Common Mistakes:

• Using non-existent methods like read_files or load
• Confusing dataset creation with file reading
• Misspelling method names

3. Given the code below, what will be the output when iterating over the dataset?

import tensorflow as tf
image_paths = ["img1.jpg", "img2.jpg"]
dataset = tf.data.Dataset.from_tensor_slices(image_paths)
for item in dataset:
    print(item.numpy().decode())

medium

A. Error: decode() not found

B. [b'img1.jpg', b'img2.jpg']

C. img1.jpg\nimg2.jpg

D. Tensor objects printed

Solution

1. Step 1: Understand dataset content

   The dataset contains string tensors of file paths: b'img1.jpg', b'img2.jpg'.

2. Step 2: Decode bytes to string

   Calling item.numpy() returns bytes, and decode() converts bytes to normal strings.

3. Final Answer:

   img1.jpg\nimg2.jpg -> Option C

4. Quick Check:

   Decoded bytes = file names [OK]

Hint: Use .numpy().decode() to get string from tensor [OK]

Common Mistakes:

• Printing tensor directly without decoding
• Expecting list output instead of individual prints
• Confusing bytes and strings

4. Identify the error in the following code snippet that tries to read image files from paths:

import tensorflow as tf
image_paths = ["img1.jpg", "img2.jpg"]
dataset = tf.data.Dataset.from_tensor_slices(image_paths)
dataset = dataset.map(tf.io.read_file)
for img in dataset:
    print(img.numpy().shape)

medium

A. Cannot print shape of a scalar string tensor

B. tf.io.read_file is not a valid function

C. from_tensor_slices requires a tensor, not list

D. map() cannot be used on datasets

Solution

1. Step 1: Analyze dataset after map

   After mapping tf.io.read_file, each element is a scalar string tensor containing raw file bytes.

2. Step 2: Understand tensor shape

   img.numpy() returns Python bytes (raw file content), which has no .shape attribute. Printing img.numpy().shape raises AttributeError.

3. Final Answer:

   Cannot print shape of a scalar string tensor -> Option A

4. Quick Check:

   img.numpy() is bytes; no .shape [OK]

Hint: Raw file bytes are scalars; no shape attribute [OK]

Common Mistakes:

• Assuming read_file returns image tensor
• Thinking from_tensor_slices rejects lists
• Believing map() is invalid on datasets

5. You want to create a TensorFlow dataset from a folder of images, resize each image to 128x128, and batch them in groups of 16. Which code snippet correctly achieves this?

hard

A. dataset = tf.keras.utils.image_dataset_from_directory('images', image_size=(128,128), batch_size=16)

B. dataset = tf.data.Dataset.list_files('images/*').map(lambda x: tf.image.resize(tf.io.decode_image(tf.io.read_file(x)), (128,128))).batch(16)

C. dataset = tf.data.Dataset.from_tensor_slices('images').map(tf.io.read_file).batch(16)

D. dataset = tf.keras.preprocessing.image_dataset_from_directory('images', batch_size=128, image_size=(16,16))

Solution

1. Step 1: Understand dataset creation from folder

   dataset = tf.data.Dataset.list_files('images/*').map(lambda x: tf.image.resize(tf.io.decode_image(tf.io.read_file(x)), (128,128))).batch(16) uses list_files to get file paths, then maps reading, decoding, and resizing images correctly.

2. Step 2: Check batch and resize parameters

   Images are resized to (128,128) and batched in groups of 16 as required.

3. Final Answer:

   dataset = tf.data.Dataset.list_files('images/*').map(lambda x: tf.image.resize(tf.io.decode_image(tf.io.read_file(x)), (128,128))).batch(16) -> Option B

4. Quick Check:

   list_files + map + resize + batch = correct pipeline [OK]

Hint: Use list_files + map with decode and resize, then batch [OK]

Common Mistakes:

• Using wrong batch size or image size parameters
• Confusing keras and tf.data APIs
• Not decoding images before resizing