Bộ dữ liệu hoa có thể được hình dung với sự trợ giúp của thư viện ‘matplotlib’. Phương thức ‘imshow’ được sử dụng để hiển thị hình ảnh trên bảng điều khiển. Toàn bộ tập dữ liệu được lặp lại và chỉ một số hình ảnh đầu tiên được hiển thị.
Đọc thêm: TensorFlow là gì và cách Keras làm việc với TensorFlow để tạo Mạng thần kinh?
Chúng tôi sẽ sử dụng tập dữ liệu về hoa, chứa hình ảnh của hàng nghìn bông hoa. Nó chứa 5 thư mục con và có một thư mục con cho mọi lớp.
Chúng tôi đang sử dụng Google Colaboratory để chạy đoạn mã dưới đây. Google Colab hoặc Colaboratory giúp chạy mã Python qua trình duyệt và không yêu cầu cấu hình cũng như quyền truy cập miễn phí vào GPU (Đơn vị xử lý đồ họa). Colaboratory đã được xây dựng trên Jupyter Notebook.
import matplotlib.pyplot as plt
print("Visualizing the flower dataset")
plt.figure(figsize=(10, 10))
for images, labels in train_ds.take(1):
for i in range(6):
ax = plt.subplot(3, 3, i + 1)
plt.imshow(images[i].numpy().astype("uint8"))
plt.title(class_names[labels[i]])
plt.axis("off")
print("Iterating over dataset")
print("Retrieving batches of images")
for image_batch, labels_batch in train_ds:
print(image_batch.shape)
print(labels_batch.shape)
break Tín dụng mã:https://www.tensorflow.org/tutorials/load_data/images
Đầu ra
Visualizing the flower dataset Iterating over dataset Retrieving batches of images (32, 180, 180, 3) (32,)
Giải thích
- Tập dữ liệu về hoa được hiển thị trực quan bằng thư viện matplotlib.
- 6 hình ảnh đầu tiên được lặp lại và hiển thị trên bảng điều khiển.
- Một lần nữa, tập dữ liệu được lặp lại và kích thước của hình ảnh được hiển thị trên bảng điều khiển.