1. Learn
  2. /
  3. Courses
  4. /
  5. Nhập môn Deep Learning với Keras

Connected

Exercise

Chuẩn bị ảnh đầu vào của bạn

Mô hình ResNet50 gốc được huấn luyện với ảnh kích thước 224 x 224 pixel và một số bước tiền xử lý; như việc trừ đi giá trị pixel trung bình của toàn bộ tập huấn luyện cho mọi ảnh huấn luyện. Bạn cần tiền xử lý các ảnh muốn dự đoán theo cùng cách đó.

Khi dự đoán trên một ảnh đơn, bạn cần đưa nó về đúng dạng đầu vào của mô hình, trong trường hợp này là: (batch-size, width, height, channels). np.expand_dims với tham số axis = 0 sẽ thêm chiều batch-size, thể hiện rằng một ảnh đơn lẻ sẽ được đưa vào để dự đoán. Giá trị cho chiều batch-size là 1, vì bạn chỉ dự đoán trên một ảnh.

Bạn sẽ đi qua các bước tiền xử lý này khi chuẩn bị ảnh chú chó (tên Ivy) để ảnh có thể được phân loại bởi ResNet50.

Instructions

100 XP
  • Import image từ tensorflow.keras.preprocessing và preprocess_input từ tensorflow.keras.applications.resnet50.
  • Tải ảnh với target_size phù hợp cho mô hình của bạn.
  • Chuyển ảnh thành mảng với image.img_to_array().
  • Tiền xử lý img_expanded theo đúng cách mà ảnh huấn luyện ResNet50 gốc đã được xử lý bằng preprocess_input().