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

Cách kiểm tra tập dữ liệu CIFAR bằng TensorFlow và Python

Việc xác minh tập dữ liệu CIFAR được thực hiện bằng cách vẽ trực quan các ảnh lên console. Vì nhãn (label) của CIFAR lưu dưới dạng mảng, cần thêm một chỉ số (index) để truy xuất đúng nhãn. Thư viện matplotlib với phương thức imshow() được dùng để hiển thị ảnh.

Mã ví dụ trên Google Colab

print("Đang kiểm tra dữ liệu")
plt.figure(figsize=(10,10))
print("Vẽ 15 ảnh đầu tiên")
print("Cần chỉ số thêm vì nhãn CIFAR là mảng")
for i in range(15):
    plt.subplot(5,5,i+1)
    plt.xticks([])
    plt.yticks([])
    plt.grid(False)
    plt.imshow(train_images[i], cmap=plt.cm.binary)
    plt.xlabel(class_names[train_labels[i][0]])
plt.show()

Nguồn mã: TensorFlow Tutorials

Đầu ra

Đang kiểm tra dữ liệu
Vẽ 15 ảnh đầu tiên
Cần chỉ số thêm vì nhãn CIFAR là mảng

Giải thích

  • Dữ liệu đã được chuẩn hóa (normalized) được trực quan hóa.
  • Sử dụng thư viện matplotlib để vẽ.