Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Tải dataset hoa (flower_photos) với TensorFlow/Keras Python

Tải dataset hoa từ Google Cloud Storage bằng tf.keras.utils.get_file.

Mã ví dụ

import matplotlib.pyplot as plt
import numpy as np
import os
import PIL
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras.models import Sequential
import pathlib

print("Import packages...")

dataset_url = "https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz"
data_dir = tf.keras.utils.get_file('flower_photos', origin=dataset_url, untar=True)
data_dir = pathlib.Path(data_dir)

print("Dữ liệu đã tải xuống")

Kết quả

Import packages...
Downloading data from https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz
228818944/228813984 [==============================] - 5s 0us/step
Dữ liệu đã tải xuống

Giải thích

  • get_file(fname, origin, untar=True): Tải file, giải nén tar.gz, trả về đường dẫn thư mục.
  • Dataset: ~228MB, 3670 ảnh, 5 class.
  • pathlib.Path: Đối tượng đường dẫn hiện đại, hỗ trợ .glob().
  • Các import: TensorFlow, Keras, layers, Sequential, matplotlib, numpy, PIL, pathlib.