Cách dùng Dataset.map trong TensorFlow tạo dataset cặp (ảnh, nhãn)

Cặp (image, label) được tạo bằng cách chuyển đổi danh sách thành phần đường dẫn (path components), sau đó mã hóa nhãn thành số nguyên. Phương thức map() giúp tạo dataset tương ứng với cặp (image, label).

Mã ví dụ

print("Set num_parallel_calls để load/xử lý nhiều ảnh song song")
train_ds = train_ds.map(process_path, num_parallel_calls=AUTOTUNE)
val_ds = val_ds.map(process_path, num_parallel_calls=AUTOTUNE)

for image, label in train_ds.take(1):
    print("Shape ảnh:", image.numpy().shape)
    print("Nhãn:", label.numpy())

Đầu ra

Set num_parallel_calls để load/xử lý nhiều ảnh song song
Shape ảnh: (180, 180, 3)
Nhãn: 0

Nguồn mã: TensorFlow Images Tutorial

Giải thích

• Nhiều ảnh được load và xử lý song song nhờ num_parallel_calls=AUTOTUNE.
• map() tạo dataset chứa cặp (image, label).
• Lặp qua dataset, in shape ảnh và nhãn.