Máy ảnh tải hình ảnh màu

Question

Tôi có 3 thư mục có hình ảnh màu. Tên của thư mục là nhãn cho các hình ảnh bên trong.

cls1 
    |____img_0.png 
    |____ ... 
    |____img_n.png 
cls2 
    |____img_0.png 
    |____ ... 
    |____img_n.png 
cls3 
    |____img_0.png 
    |____ ... 
    |____img_n.png 

Tôi muốn sử dụng thư viện Keras để tạo mạng nơron để phân loại, nhưng tôi không thể tìm thấy, cách tạo tập dữ liệu từ hình ảnh màu. Bạn có thể giúp tôi không?

Answer 1

Vui lòng xem xét gist of pre-trained VGG-16 model with example usage này mà tôi tìm thấy minh họa:

Để tóm tắt:

1. Tải hình ảnh của bạn với OpenCV hoặc scikit hình ảnh, vv (L58)
2. Resize và/hoặc cắt của bạn hình ảnh để vừa với kích thước đầu vào (224 * 224 cho VGG-16) (L58)
3. Tính và trừ cho mỗi hình ảnh giá trị trung bình (L59 ~ L61)
4. Hoán đổi kích thước màu trước chiều cao và kích thước chiều rộng (L62)
5. Nếu bạn sử dụng scikit-hình ảnh, bạn cần phải trao đổi các kênh màu 3 vì tải OpenCV hình ảnh như BGR kênh nhưng scikit hình ảnh tải nó như RGB kênh.
6. Thêm kích thước kích thước hàng loạt (L63)
7. Shuffle, phân vùng và tiếp nhau chúng (cùng chiều 0) để tạo thành dữ liệu huấn luyện X_train, dữ liệu thử nghiệm X_test vv cùng với mặt đất thật Y_train, Y_test, vv
8. Nếu bộ dữ liệu của bạn quá lớn để vừa với bộ nhớ, hãy sử dụng trình tạo và chức năng fit_generator để thực hiện việc đào tạo thay thế. (Keras cũng có evaluate_generator và predict_generator)
9. Bạn đã sẵn sàng để được đào tạo.