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

Khám phá dataset hoa với Keras Sequential API Python

Duyệt qua thư mục dataset để đếm số ảnh và xem trước một vài hình ảnh mẫu.

Mã ví dụ

import pathlib
import PIL

image_count = len(list(data_dir.glob('*/*.jpg')))
print("Số ảnh trong dataset:", image_count)

print("\nXem trước dataset:")
print("ROSES")
roses = list(data_dir.glob('roses/*'))
display(PIL.Image.open(str(roses[1])))

print("TULIPS")
tulips = list(data_dir.glob('tulips/*'))
display(PIL.Image.open(str(tulips[0])))

Kết quả

Số ảnh trong dataset: 3670
Xem trước dataset:
ROSES
TULIPS

Khám phá dataset hoa với Keras Sequential API Python Khám phá dataset hoa với Keras Sequential API Python

Giải thích

  • data_dir.glob('*/*.jpg'): Đệ quy tìm tất cả file .jpg trong thư mục con.
  • PIL.Image.open(): Mở ảnh để hiển thị (trong Jupyter/Colab dùng display()).
  • Dataset: 3670 ảnh, 5 class, cấu trúc thư mục chuẩn cho image_dataset_from_directory.