Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

TensorFlow: ดาวน์โหลด Flower Dataset ด้วย Keras Sequential API

ดาวน์โหลด flower dataset จาก Google Cloud Storage ด้วย tf.keras.utils.get_file

โค้ดตัวอย่าง

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("นำเข้าแพ็กเกจที่จำเป็น...")

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("ดาวน์โหลดข้อมูลเสร็จสิ้น")

ผลลัพธ์

นำเข้าแพ็กเกจที่จำเป็น...
Downloading data from https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz
228818944/228813984 [==============================] - 5s 0us/step
ดาวน์โหลดข้อมูลเสร็จสิ้น

คำอธิบาย

  • get_file(fname, origin, untar=True): ดาวน์โหลดไฟล์จาก URL, แตกไฟล์ .tgz/.tar.gz อัตโนมัติ
  • คืนค่า path ของไดเรกทอรีที่แตกไฟล์
  • pathlib.Path: จัดการ path แบบ object-oriented (cross-platform)
  • Dataset: 3670 รูปภาพ 5 คลาส ประมาณ 228 MB
  • เป็นขั้นตอนแรกก่อน preprocessing และสร้างโมเดล